Spaces:
Sleeping
Sleeping
update app.py (fast_hdbscan)
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ import numpy as np
|
|
19 |
from sklearn.decomposition import PCA
|
20 |
from sklearn.cluster import KMeans
|
21 |
import umap
|
22 |
-
import
|
23 |
import plotly.graph_objects as go
|
24 |
import plotly.express as px
|
25 |
from sklearn.feature_extraction.text import TfidfVectorizer, CountVectorizer
|
@@ -196,7 +196,7 @@ class UAPAnalyzer:
|
|
196 |
"""
|
197 |
logging.info(f"Clustering data using {method}")
|
198 |
if method == 'HDBSCAN':
|
199 |
-
clusterer =
|
200 |
elif method == 'KMeans':
|
201 |
clusterer = KMeans(**kwargs)
|
202 |
else:
|
|
|
19 |
from sklearn.decomposition import PCA
|
20 |
from sklearn.cluster import KMeans
|
21 |
import umap
|
22 |
+
import fast_hdbscan
|
23 |
import plotly.graph_objects as go
|
24 |
import plotly.express as px
|
25 |
from sklearn.feature_extraction.text import TfidfVectorizer, CountVectorizer
|
|
|
196 |
"""
|
197 |
logging.info(f"Clustering data using {method}")
|
198 |
if method == 'HDBSCAN':
|
199 |
+
clusterer = fast_hdbscan.HDBSCAN(**kwargs)
|
200 |
elif method == 'KMeans':
|
201 |
clusterer = KMeans(**kwargs)
|
202 |
else:
|