2013-05-06 35 views
11

DecisionTreeClassifier'ın kriter = 'entropy' kabul ettiğini görüyorum, bu da karar ağacını ayırmak için bir ölçüt olarak bilgi kazancı kullanması gerektiği anlamına geliyor. İhtiyacım olan şey, kök düğümünü bölmek üzere olduğunda, her özellik için kök düzeyinde bilgi kazanımıdır.Bir bilgi edinmek için nasıl bilgi edinilir DecisionTreeClassifier?

cevap

7

Yalnızca bölme düğümü olarak kullanılan bir özellik için bilgi kazancına (veya gini kirliliğine) erişebilirsiniz. DecisionTreeClassifier.tree_.best_error[i] özniteliği, DecisionTreeClassifier.tree_.feature[i] özelliğinde i-th düğüm bölmesinin entropisini tutar. Eğer i-th düğümüne ulaşan tüm örneklerin entropisini DecisionTreeClassifier.tree_.init_error[i]'a bakın. Burada belgelere bakın Daha fazla bilgi için

: https://github.com/scikit-learn/scikit-learn/blob/dacfd8bd5d943cb899ed8cd423aaf11b4f27c186/sklearn/tree/_tree.pyx#L64

Eğer (belirli bölünmüş düğümde) her özellik için entropi erişmek istiyorsanız - Benim bilmediğim fonksiyonunu find_best_splithttps://github.com/scikit-learn/scikit-learn/blob/master/sklearn/tree/_tree.pyx#L713

+0

değiştirmeniz gerekir Github hakkında belgelerin olduğunu anlayın. Teşekkürler. –

İlgili konular