2016-12-09 1 views
0

Ich verwende Sklearn RFC.Random Forest Classifier: Merkmal Wichtigkeit der Vorhersage Wahrscheinlichkeit

forest.fit(training_data, y_train) 
probas_test = forest.predict_proba(test_data) 

I gibt es eine Möglichkeit wissen, wollte den Beitrag/Bedeutung der einzelnen Funktionen zu finden, die zu der Vorhersage führen.

etwas wie, aber für eine individuelle Datenpunkt-Ebene.

forest.feature_importances_ 
+0

ist nicht '.features_importances_' zurück geben, was Sie wollen? – MMF

+0

Danke für die Antwort @MMF. Es wird für den vollständig trainierten Baum zurückgegeben, während ich diese Parameter für die individuelle Vorhersage von zu beschreibenden Daten suche. – OneWorld

Antwort

0

Dies kann auf viele Arten gelöst werden; Überprüfen Sie http://blog.datadive.net/interpreting-random-forests/ (und ein Python-Paket dafür: https://github.com/andosa/treeinterpreter). Es gibt auch weniger direkte Optionen, z.B.

+0

Das ist genau das, was ich gesucht habe. Vielen Dank ! – OneWorld

Verwandte Themen