Как установить имя столбца в древовидной структуре через tree.plot_tree?

Question

Я пытаюсь нарисовать дерево через библиотеку sklearn lib, но проблема в том, что индексы столбцов написаны на рисунке.

tree.plot_tree(clf_decision)

Answer 1

использовать feature_names и class_names параметры :

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)