小编Woc*_*ush的帖子

我目前有一个决策树，将功能名称显示为X[index]、 ieX[0], X[1], X[2]等。

from sklearn import tree
from sklearn.tree import DecisionTreeClassifier

dt = DecisionTreeClassifier()
dt.fit(X_train, y_train)

 
# plot tree
plt.figure(figsize=(20,16))# set plot size (denoted in inches)
tree.plot_tree(dt,fontsize=10)

我希望将这些 X[featureNumber] 替换为实际的功能名称。

因此，我希望它显示返回的功能名称，而不是显示 X[0] X.columns.values[0]（我不知道此代码是否正确）。

我也知道有一种使用 graphviz 来完成此操作的简单方法，但由于某种原因我无法让 graphviz 在 Jupiter 中运行，所以我正在寻找一种不使用 graphviz 的方法。

当前决策树的照片：