Using scikit-learn DecisionTreeClassifier to cluster

Question

When using sklearn.tree.DecisionTreeClassifier, the classifier has methods for predicting probability and class.

Is there a way to use the same tree for clustering: for a given input vector x, simply tell which leaf x belongs to?

Answer 1

I found the answer to my own question - leaving it here as reference for the next time someone looks for it:

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