使用scikit-learn decisiontreecclassifier进行聚类



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

相关内容

  • 没有找到相关文章

最新更新