Alternative to scipy.cluster.hierarchy.cut_tree()
One way to obtain k
flat clusters is to use scipy.cluster.hierarchy.fcluster
with criterion='maxclust'
:
from scipy.cluster.hierarchy import fclusterclust = fcluster(Z, k, criterion='maxclust')