Spaces:
Running
Running
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.
|
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()
|