2015-12-11 4 views
7

Meine Frage ist, können wir wählen, welche Decision Tree-Algorithmus in Sklearn verwendet werden?Können wir wählen, welcher Decision Tree-Algorithmus in sklearn verwendet werden soll?

In der Bedienungsanleitung von sklearn erwähnt es optimierte Version des CART-Algorithmus verwendet wird.

Können wir zu anderen Algorithmen wie C4.5 wechseln?

+0

Was genau meinen Sie damit? Fehlende Werte und kategorische Variablen werden derzeit nicht behandelt, und es gibt kein Post-Pruning, nur Pre-Pruning. C4.5 ist nicht wirklich ein Algorithmus, es ist ein Programm, oder? –

Antwort

5

Nr Siehe documentation

scikit-learn uses an optimised version of the CART algorithm. 
0

Aber es gibt ein params Kriterium, das wir wählen können "Gini" oder "Entropie" verwenden:

clf = tree.DecisionTreeClassifier(criterion="entropy") 

Kriterium: string, optional (default = "gini") Die Funktion die Qualität eines Split zu messen. Unterstützte Kriterien sind "Gini" für die Gini Verunreinigung und "Entropie" für den Informationsgewinn.

siehe Docs