Spaces:
Sleeping
Sleeping
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=2, metric='cosine', n_jobs=-1)
|
| 11 |
embeddings_array = np.array(embeddings)
|
| 12 |
labels = dbscan.fit_predict(embeddings_array)
|
| 13 |
labels = labels.tolist()
|