Graphviz决策树输出未显示标准/基尼

Question

我想基于决策树模型输出显示Graphviz决策树图像，因为它更可表达，但是，来自初始模型输出的信度值'gini'或'熵'并未显示在graphviz树上输出。

我遵循了本教程：https://www.datacamp.com/community/tutorials/decision-tree-classification-python

用于决策树输入的代码：

clf = tree.DecisionTreeClassifier(max_leaf_nodes = 3, min_samples_leaf = 5, max_depth =4)
clf = clf.fit(X_train, y_train)
tree.plot_tree(clf.fit(X_train, y_train))

决策树模型的输出：https://i.stack.imgur.com/NprzI.png

用于graphviz输入的代码：

dot_data = StringIO()
tree.export_graphviz(clf, out_file = dot_data, feature_names = inputs.columns,class_names =['0','1'], filled = True, rounded = True, impurity = False)

graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
Image(graph.create_png())

graphviz决策树的输出：https://i.stack.imgur.com/5Q9D6.png

我在clf树的参数中添加了criteria =“ gini”（尽管由于默认情况下将其设置为gini，所以不是必需的，但未对graphviz输出进行更改。

我还在clf树的参数中添加了标准=“熵”，该参数将输出从gini更改为熵，并显示在树模型输出上，而不显示在graphviz输出上。

我在文档或其他地方没有看到任何东西可以说明为什么会出现这种情况，并且对于显示所使用的标准很有用。

我在某处缺少参数了吗？

Answer 1

摘自sklearn.tree.export_graphviz的文档：