想问下,plot决策树时的class_name枚举值顺序,是如何确定的呢?

来源:5-6 实战(1)

tutu17

2020-03-25

想问下,plot决策树时的class_name枚举值顺序,是如何确定的呢?

写回答

1回答

flare_zhao

2020-03-25

同学你好,决策树的分支原则是按照比如老师课程中提到的ID3算法进行确定。在具体进行画图的时候,我们可以通过feature_names和class_names告诉决策树对应的feature和label的名称,这样就可以实现1、顺序的确定(通过ID3算法)2、名称的匹配

tree.plot_tree(dc_tree,filled='True',feature_names=['SepalLength', 'SepalWidth', 'PetalLength', 'PetalWidth'],class_names=['setosa','versicolor','virginica'])


0
6
flare_zhao
回复
码农小陈
加油哈
2020-08-07
共6条回复

Python3入门人工智能 掌握机器学习+深度学习

人工智能基础全方位讲解,构建完整人工智能知识体系,带你入门AI

1946 学习 · 594 问题

查看课程