我爱编程

机器学习之决策树

2018-07-26  本文已影响1人  kate2020

刚刚在kaggle上练习了决策树的回归训练, 预测房价
决策树图示:


7tsb5b1.png

决策树通过分类的方法得到样本的预测值, 叶子结点的值就是该分类的平均值
需要注意的是决策树的深度, 过浅容易造成欠拟合, 过深容易造成过拟合
所以随机森林相比决策树更有优势

上一篇 下一篇

猜你喜欢

热点阅读