使用scikit-learn DecisionTreeClassifier进行聚类

Question

使用sklearn.tree.DecisionTreeClassifier时,分类器具有预测概率和类的方法.

有没有办法使用相同的树进行聚类:对于给定的输入向量x,只需告诉哪个叶子x属于哪个？

Answer 1

我找到了自己的问题的答案 - 将其留在这里作为下次有人寻找时的参考:

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)))