Eurosmart commited on
Commit
bd8a530
·
1 Parent(s): 3812714
Files changed (1) hide show
  1. api/routes/clusterRoute.py +1 -1
api/routes/clusterRoute.py CHANGED
@@ -7,7 +7,7 @@ def cluster():
7
  from sklearn.cluster import DBSCAN
8
  import numpy as np
9
  try:
10
- dbscan = DBSCAN(eps=0.1, min_samples=7, metric='cosine', n_jobs=-1)
11
  embeddings_array = np.array(embeddings)
12
  labels = dbscan.fit_predict(embeddings_array)
13
  labels = labels.tolist()
 
7
  from sklearn.cluster import DBSCAN
8
  import numpy as np
9
  try:
10
+ dbscan = DBSCAN(eps=0.2, min_samples=5, metric='cosine', n_jobs=-1)
11
  embeddings_array = np.array(embeddings)
12
  labels = dbscan.fit_predict(embeddings_array)
13
  labels = labels.tolist()