使用管道机制Pipeline封装岭回归模型

利用scikit-learn的Pipeline,封装(生成特征多项式、数据归一化、岭回归模型构建)流程。

pipe=Pipeline([
       ( 'poly',PolynomialFeatures()),
        ('scaler',StandardScaler()),
        ('lin',Ridge())
    ])

用GridSearchCV,对Pipeline中PolynomialFeatures的degree参数、和岭回归模型的alpha参数进行调优.

param_grid=[{'poly__degree':[i for i in range(1,6)],'lin__alpha':[i / 10 for i in range(1,20)]}]
#使用网格搜索,并设置r2指标评估
gs_pipe=GridSearchCV(pipe,param_grid,scoring='r2')

接下来,进行数据训练,就可以找到最优模型的参数了。

gs_pipe.fit(x_train,y_train)
gs_pipe.best_params_
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。