fix
Browse files
modular_graph_and_candidates.py
CHANGED
|
@@ -156,11 +156,14 @@ def embedding_similarity_clusters(models_root: Path, missing: List[str], thr: fl
|
|
| 156 |
sims_mat = embeddings @ embeddings.T
|
| 157 |
|
| 158 |
out = {}
|
| 159 |
-
|
| 160 |
-
|
|
|
|
|
|
|
|
|
|
| 161 |
s = float(sims_mat[i, j])
|
| 162 |
if s >= thr:
|
| 163 |
-
out[(
|
| 164 |
return out
|
| 165 |
|
| 166 |
|
|
|
|
| 156 |
sims_mat = embeddings @ embeddings.T
|
| 157 |
|
| 158 |
out = {}
|
| 159 |
+
matrix_size = embeddings.shape[0] # Actual number of embeddings we have
|
| 160 |
+
processed_names = names[:matrix_size] # Only use names that have embeddings
|
| 161 |
+
|
| 162 |
+
for i in range(matrix_size):
|
| 163 |
+
for j in range(i + 1, matrix_size):
|
| 164 |
s = float(sims_mat[i, j])
|
| 165 |
if s >= thr:
|
| 166 |
+
out[(processed_names[i], processed_names[j])] = s
|
| 167 |
return out
|
| 168 |
|
| 169 |
|