python - 决策树显示没有达到我想要的效果

Question

我正在尝试正确显示决策树；我非常接近，但我就是无法正确显示类(class)。

当我使用 class_names=True 时，我得到:

如何解释 class=y[1]？

这里有更多详细信息:

和

请注意，图像被剪裁，但它仍沿着相同的线条继续。

然后:

clf.fit(X, Y)
dot_data = tree.export_graphviz(clf, out_file=None, feature_names=combo.columns[2:], class_names=class_names)
graph = graphviz.Source(dot_data)
graph.render('r')

它会产生这样的输出，我知道这是错误的，因为唯一出现 5 次的成绩是 A。

问题是什么？

相关链接

https://datascience.stackexchange.com/questions/28574/decisiontreeclassifier-object-has-no-attribute-importances这是一个有趣的答案，其中包含一些有用的提示，但根本没有回答我在这里提出的问题。

Answer 1

我找到了这个解决方案:

clf = tree.DecisionTreeClassifier(max_depth=4)
clf.fit(X, Y)
dot_data = tree.export_graphviz(clf, out_file=None, feature_names=Xprep.columns, class_names=clf.classes_)

事实证明，在 fit 调用的结果上有一个稍微难以找到的属性:clf.classes_，其中包含类的文本再现。