diff --git a/deeptools/heatmapper.py b/deeptools/heatmapper.py index fb9b9139c..a7b65c6a1 100644 --- a/deeptools/heatmapper.py +++ b/deeptools/heatmapper.py @@ -1077,8 +1077,10 @@ def hmcluster(self, k, method='kmeans'): if method == 'hierarchical': # normally too slow for large data sets - from scipy.cluster.hierarchy import fclusterdata - cluster_labels = fclusterdata(matrix, k, criterion='maxclust', metric='euclidean', depth=2, method='ward') + from scipy.cluster.hierarchy import fcluster, linkage + Z = linkage(matrix, method='ward', metric='euclidean') + cluster_labels = fcluster(Z, k, criterion='maxclust') + # create groups using the clustering self.group_labels = [] self.group_boundaries = [0]