热线电话:13121318867

登录
2020-08-09 阅读量: 1284
机器学习回归树的构建以及参数的优化,和模型的保存具体操作流程

1.png

上图是回归树的代码操作,只是简单的执行,里面的参数并没有优化,

2.png

上图是对回归树的一些参数进行优化的操作,图中的1,是可以显示出模型在优化的过程当然也可以不设置,2是打印出最优模型形成的参数,3可以说是相同的意义取一个就可以,最下面的是将打印出的最优参数进行带入测试,而上面是直接在2的基础上得分。

3.png

上图是如果觉得这个模型比较好还需要再利用,可以进行保存,可以节约下次使用模型还需要进行训练的流程,把直接训练好的模型,使用from sklearn.externals import joblib来进行保存,后面是要保存的路径,joblib.dump(reg,r'C:\Users\Administrator\Desktop\4days-1\3.决策树\reg.pkl') 其中红色字的是要保存的模型名称,绿色字的是保存在路径下的模型名称,可以换成别的名,但是需要是pkl的格式。以下2,3皆有解释。

31.4607
1
关注作者
收藏
评论(0)

发表评论

暂无数据
推荐帖子