在使用sklearn.tree.DecisionTreeClassifier时,分类器具有预测概率和类别的方法。
是否有办法使用同一棵树进行聚类:对于给定的输入向量x,简单地告诉x属于哪个叶子节点?
回答:
我找到了自己的问题的答案 – 留在这里作为下次有人寻找时的参考:
import numpy as npimport sklearn.treeclf = sklearn.tree.DecisionTreeClassifier()clf.fit(X,y)clf.tree_.apply(np.asfortranarray(X.astype(sklearn.tree._tree.DTYPE)))