TPOT自动选择机器学习模型和参数--回归示例

前两篇博客写了在anacoda下安装tpot库和使用tpot做分类的例子，这篇是写做回归的例子

anacoda下安装tpot库

使用TPOT自动选择scikit-learn机器学习模型和参数--分类示例

环境：win10+pycharm+anacoda

数据集：sklearn自带的波士顿房价数据集

代码：

'''
    回归，预测波士顿房价
    '''
    from tpot import TPOTRegressor
    import pandas as pd
    import numpy as np
    from sklearn.datasets import load_boston
    from sklearn.model_selection import train_test_split
 
    housing = load_boston()
    X_train, X_test, y_train, y_test = train_test_split(housing.data, housing.target)
                                                       # ,train_size=0.75, test_size=0.25)

    tpot = TPOTRegressor(generations=20, verbosity=2) #迭代20次
    tpot.fit(X_train, y_train)