如何通过tree.plot_tree设置树视图中的列名？

Posted 2023-03-12

【中文标题】如何通过tree.plot_tree设置树视图中的列名？

【英文标题】：How to set the name of columns in the tree view through tree.plot_tree?

【发布时间】：2019-10-23 11:49:25

【问题描述】：

我正在尝试通过sklearn lib树来画一棵树，但问题是图中写了列索引。

tree.plot_tree(clf_decision)

【参考方案1】：

利用feature_names and class_names parameters:

from sklearn.datasets import load_iris
from sklearn import tree

iris = load_iris()
clf = tree.DecisionTreeClassifier(random_state=0).fit(iris.data, iris.target)


tree.plot_tree(clf, feature_names=iris.feature_names, class_names=iris.target_names)