我正在使用scikit learn在Python上使用Decision Trees做一些工作。分类器本身工作得非常好，但是当我创建这个的图形表示时，不是显示变量名，而是显示变量的X [0]，X [1] ... X [n]，就像在scikit中一样学习示例：http://scikit-learn.org/stable/modules/tree.html。让事情变得更加困难的是X [0]甚至不一定与第一列相关，X [1]与第二列相关，等等，似乎在分配我的变量时会有一些过程在进行新的与X相关的＆＃39;名。
如何将变量名称放在图形表示上，将原始变量名称映射到新的变量名称，或确保新变量名称按顺序排列？
答案 0 :(得分：0)
有tree.export_graphviz()方法的参数称为feature_names来指定要素名称，document说：
feature_names : list of strings, optional (default=None)
Names of each of the features.
要素名称的顺序是您训练模型的要素向量中要素的顺序。