6
使用sklearn.tree.DecisionTreeClassifier時,分類器具有預測概率和類的方法。使用scikit-learn DecisionTreeClassifier集羣
有沒有辦法使用相同的樹進行聚類:對於給定的輸入向量x,只需告訴哪個葉子屬於?
使用sklearn.tree.DecisionTreeClassifier時,分類器具有預測概率和類的方法。使用scikit-learn DecisionTreeClassifier集羣
有沒有辦法使用相同的樹進行聚類:對於給定的輸入向量x,只需告訴哪個葉子屬於?
我找到了答案,以我自己的問題 - 在這裏把它當作下一次有人查找它以供參考:
import numpy as np
import sklearn.tree
clf = sklearn.tree.DecisionTreeClassifier()
clf.fit(X,y)
clf.tree_.apply(np.asfortranarray(X.astype(sklearn.tree._tree.DTYPE)))
您還可以標記自己的答案接受。 – ogrisel
謝謝。原來你需要等待兩天才能接受你自己的答案:) –
這絕對是我們需要更多文檔的東西。順便說一句,你可能會對RandomTreesEmbedding感興趣:http://scikit-learn.org/dev/modules/generated/sklearn.ensemble.RandomTreesEmbedding.html#sklearn.ensemble.RandomTreesEmbedding森林也直接具有「應用」功能。 –