Python決策樹 - 創建包含變量名稱的決策樹的圖形表示

Question

我正在使用scikit學習在Python上使用決策樹進行一些工作。分類器本身工作得很好，但是當我創建一個圖形化表示形式時，不是顯示變量名，而是顯示變量的X [0]，X [1] ... X [n]，如scikit學習示例：http://scikit-learn.org/stable/modules/tree.html。更重要的是，X [0]不一定與第一列X [1]到第二列等有關，似乎有一些過程將我的變量分配給新的'X相關'名。

我該如何在圖形表示上放置變量名稱，將我的原始變量名稱映射到新的變量名稱，還是確保新的變量名稱按順序排列？

Answer 1

有一個叫feature_names指定的功能名稱tree.export_graphviz()方法的說法，document說：

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

的功能名稱順序的特徵，特徵向量，你訓練模型的順序。