From 4c93e8db75fe456cd58dc15523f79eebf24c5881 Mon Sep 17 00:00:00 2001
From: DIANE <abderrahim.diane@cefe.cnrs.fr>
Date: Mon, 14 Oct 2024 12:33:54 +0200
Subject: [PATCH] HDBSCAN
---
src/utils/clustering.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/src/utils/clustering.py b/src/utils/clustering.py
index d367d04..3433ec9 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)
--
GitLab