机器学习之分类树实现

2025-10-24 22:57:28

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、以上就是实现分类树的全过程

声明:本网站引用、摘录或转载内容仅供网站访问者交流或参考,不代表本站立场,如存在版权或非法内容,请联系站长删除,联系邮箱:site.kefu@qq.com。
猜你喜欢