必备！人工智能和数据科学的七大 Python 库( 九 )_图1

1fromtpotimportTPOTClassifier

2fromsklearn.datasetsimportload_digits

3fromsklearn.model_selectionimporttrain_test_split

5# load and split dataset

6digitsdigits == load_digitsload_di ()

7X_train, X_test, y_train, y_test = train_test_split(digits.data, digits.target,

8train_size=0.75, test_size=0.25)

10# Fit the TPOT classifier

11tpot = TPOTClassifier(verbosity=2, max_time_mins=5, population_size=40)

12tpot.fit(X_train, y_train)

14# Export pipeline

15tpot.export('tpot_mnist_pipeline.py')

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