HDBSCAN

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)