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

Question

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

Answer 1

flare_zhao 2020-03-25 23:17:46

同学你好，决策树的分支原则是按照比如老师课程中提到的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'])

• 无限Infantry #1

  老师您好， 我想追加这个问题。 就是feature_names, class_names 里面元素的排序是否会影响plt 显示的顺序，如果会影响，怎么确定这两个list元素的顺序？

• flare_zhao 回复 无限Infantry #2

  这个问题的话，你试一下体验一下不就清楚了吗。通常来说，如果可以自己快速验证的东西，自己试试，那样也会理解的更深的哈

• 码农小陈 回复 flare_zhao #3

  老师您好 我在官方的技术手册中查到了关于class_name的阐述 “Names of each of the target classes in ascending numerical order” 然后我自己也实践了一下 排序确实会影响显示顺序 所以我的理解是 'setosa','versicolor','virginica' 是根据class label的升序排列与 0,1,2 一一对应 再比如 如果数据集中的class label是 0,2,4 那么class_names 中的第一个元素对应0, 第二个对应2, 第三个对应4 
  请问老师我的理解有问题吗 没问题的话也能给有疑问的同学做个参考

点开查看后面3条评论