小朱520

2020-08-09   阅读量: 931

机器学习

机器学习回归树的构建以及参数的优化,和模型的保存具体操作流程

扫码加入数据分析学习群

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皆有解释。

添加CDA认证专家【维克多阿涛】,微信号:【cdashijiazhuang】,提供数据分析指导及CDA考试秘籍。已助千人通过CDA数字化人才认证。欢迎交流,共同成长!
31.4607 2 1 关注作者 收藏

评论(0)


暂无数据

推荐课程

推荐帖子