'Die Länge der Feature-Namen stimmt nicht mit der Anzahl der Features überein, während der Entscheidungsbaum mit Graphviz gezeichnet wird

Question

Ich versuche, einen Entscheidungsbaum mit graphviz zu visualisieren, und während ich versuche, den Entscheidungsbaum zu plotten, erhalte ich den folgenden Fehler:

~\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, 

Mein Code

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)) 

können Sie mir sagen, was falsch gelaufen?

Answer 1

Ihre data.columns enthält Namen aller Features + Label, da sie nicht in X_train_res und y_train_res unterteilt ist. Sie müssen die Feature-Namen in X_train_res anstelle von data.columns übergeben, um die genauen Features zu erhalten, ansonsten werden auch die Beschriftungen einbezogen. Ich gehe davon aus, dass X_train und y_train in diesem Fall von data abgeleitet sind.