Python 决策树 - 创建包含变量名称的决策树的图形表示

Question

我正在使用 scikit learn 在 Python 上使用决策树做一些工作。分类器本身工作得很好，但是当我创建它的图形表示时，不是显示变量名称，而是为变量显示 X[0]、X[1]...X[n]，就像在 scikit 中一样学习示例：http://scikit-learn.org/stable/modules/tree.html。更困难的是，X[0] 甚至不一定与第一列相关，X[1] 与第二列等相关，似乎在为我的变量分配新的“X 相关”时正在进行一些过程名字。

我怎样才能将变量名放在图形表示上，将我的原始变量名映射到新的对应项，或者确保新的变量名是有序的？

Answer 1

tree.export_graphviz() 方法中有一个参数称为 feature_names 来指定功能名称，document 说：

feature_names : list of strings, optional (default=None)
Names of each of the features.

特征名称的顺序是你训练模型的特征向量中特征的顺序。