diff --git a/src/utils/clustering.py b/src/utils/clustering.py index d367d042d3ce61fb78be16703dab7b24f41cfaaa..3433ec9c59d2d343d9f7874d549b65b0cacbf7c8 100644 --- a/src/utils/clustering.py +++ b/src/utils/clustering.py @@ -77,6 +77,8 @@ class Hdbscan: """ def __init__(self, data): + from sklearn.cluster import HDBSCAN + """Initiate the HDBSCAN calculation Args: @@ -405,6 +407,7 @@ class AP: self.__x = np.array(X) # Fit PCA model + from sklearn.cluster import AffinityPropagation self.M = AffinityPropagation(damping=0.5, max_iter=200, convergence_iter=15, copy=True, preference=None, affinity='euclidean', verbose=False, random_state=None) self.M.fit(self.__x)