![](/img/trans.png)
[英]Scikit-learn using GridSearchCV on DecisionTreeClassifier
[英]Using scikit-learn DecisionTreeClassifier to cluster
使用sklearn.tree.DecisionTreeClassifier時,分類器具有預測概率和類的方法。
有沒有辦法使用相同的樹進行聚類:對於給定的輸入向量x,只需告訴哪個葉子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)))
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.