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