簡體   English   中英

使用scikit-learn 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.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM