必备！人工智能和数据科学的七大 Python 库( 六 )

2019-01-05

1fromtpotimportTPOTClassifier

2fromsklearn.datasetsimportload_iris

3fromsklearn.model_selectionimporttrain_test_split

4

5# Load iris dataset

6iris = load_iris()

7

8# Split the data

9

10X_trainX_train, X_test, y_train, y_test = train_test_split(iris.data, iris.target,

11train_size=0.75, test_size=0.25)

12

13# Fit the TPOT classifier

14

15tpot = TPOTClassifier(verbosity=2, max_time_mins=2)

16tpot.fit(X_train, y_train)

推荐阅读

上一篇：思必驰AI芯片发布：内置完整语音交互方案

下一篇：女神被打码了？一笔一划脑补回来，效果超越Adobe