2020-08-09
阅读量:
1284
机器学习回归树的构建以及参数的优化,和模型的保存具体操作流程
上图是回归树的代码操作,只是简单的执行,里面的参数并没有优化,
上图是对回归树的一些参数进行优化的操作,图中的1,是可以显示出模型在优化的过程当然也可以不设置,2是打印出最优模型形成的参数,3可以说是相同的意义取一个就可以,最下面的是将打印出的最优参数进行带入测试,而上面是直接在2的基础上得分。
上图是如果觉得这个模型比较好还需要再利用,可以进行保存,可以节约下次使用模型还需要进行训练的流程,把直接训练好的模型,使用from sklearn.externals import joblib来进行保存,后面是要保存的路径,joblib.dump(reg,r'C:\Users\Administrator\Desktop\4days-1\3.决策树\reg.pkl') 其中红色字的是要保存的模型名称,绿色字的是保存在路径下的模型名称,可以换成别的名,但是需要是pkl的格式。以下2,3皆有解释。






评论(0)


暂无数据
推荐帖子
0条评论
0条评论
0条评论