机器学习之分类树实现

时间:2026-02-12 13:35:01

1、导入相关机器学习模快

机器学习之分类树实现

2、实例化,分割数据集

机器学习之分类树实现

3、建立分类树模型

机器学习之分类树实现

4、绘制分类树

机器学习之分类树实现

5、显示绘制的分类树

机器学习之分类树实现

6、绘制分类树参数说明:

1.在实例化绘制分类树对象时,需将out_file置为None,否者会有警告。

在绘制分类树后生成一个.dot的文件

机器学习之分类树实现

7、绘制分类树参数说明:

2.filled用于给节点填充颜色,默认False不填充颜色

机器学习之分类树实现

机器学习之分类树实现

8、绘制分类树参数说明:

3.rounded用于设置节点形状

默认为False每个节点的形状为矩形,True为显示为园矩形

机器学习之分类树实现

机器学习之分类树实现

9、分类树的剪枝:

max_depth 属性限制分类树的深度(不包含跟节点)

机器学习之分类树实现

机器学习之分类树实现

10、分类树的剪枝:

min_samples_leaf 属性

限定一个节点在分枝后的每个子节点都必须包含至少min_samples_leaf个训练样本

机器学习之分类树实现

机器学习之分类树实现

11、以上就是实现分类树的全过程

© 2026 海能知识库
信息来自网络 所有数据仅供参考
有疑问请联系站长 site.kefu@gmail.com