Ashoka74 commited on
Commit
40e0215
·
verified ·
1 Parent(s): 76373d4

update app.py (fast_hdbscan)

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 hdbscan
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 = hdbscan.HDBSCAN(**kwargs)
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: