我试图通过sklearn库中的tree绘制一棵树,但问题是图片中显示的是列索引。
tree.plot_tree(clf_decision)
回答:
使用feature_names
和 class_names
参数:
from sklearn.datasets import load_irisfrom sklearn import treeiris = 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)