'Graphvizを使用して意思決定ツリーをプロットする際に、フィーチャ名の長さがフィーチャの数と一致しません。

Question

graphvizを使用して意思決定ツリーを視覚化しようとしていますが、意思決定ツリーをプロットしようとしています。

~\AppData\Local\Continuum\anaconda3\lib\site-packages\sklearn\tree\export.py in export_graphviz(decision_tree, out_file, max_depth, feature_names, class_names, label, filled, leaves_parallel, impurity, node_ids, proportion, rotate, rounded, special_characters, precision) 
    427         "does not match number of features, %d" 
    428         % (len(feature_names), 
--> 429          decision_tree.n_features_)) 
    430 
    431   # The depth of each node for plotting with 'leaf' option 

ValueError: Length of feature_names, 225 does not match number of features, 

私のコード

dt=DecisionTreeClassifier(class_weight="balanced", min_samples_leaf=30) 
fit_decision=dt.fit(X_train_res,y_train_res) 
from graphviz import Source 
from sklearn import tree 
Source(tree.export_graphviz(fit_decision, out_file=None, feature_names=data.columns)) 

あなたは何が悪かったのかを教えてもらえますか？

Answer 1

あなたのdata.columnsには、X_train_resとy_train_resに分割されていないため、すべての機能+ラベルの名前が含まれています。正確な機能を得るには、data.columnsの代わりにX_train_resのフィーチャー名を渡す必要があります。この場合、X_trainとy_trainはdataから派生していると仮定しています。