python使用tpot做分类

topt做分类的简单使用

import numpy as np
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from tpot import TPOTClassifier

#### 设置随机种子
np.random.seed(10)
#### 导入数据
iris = load_iris()
X_train,X_test,y_train,y_test = train_test_split(iris.data, iris.target,
                                                    train_size=0.75, test_size=0.25)
#### generation: pipeline的循环数一般越多越好，population_size：每一个循环数中GP中的个体数，一般越多越好
tpot = TPOTClassifier(generations=5, population_size=20, verbosity=1)
tpot.fit(X_train, y_train)
print(tpot.score(X_test, y_test))