debug embeddings

ARBML · Apr 6, 2024 · 8f2f624 · 8f2f624
1 parent fbc8125
commit 8f2f624
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/utils/clusters_utils.py b/utils/clusters_utils.py
@@ -20,6 +20,8 @@ def compute_clusters(embeddings: List[List[float]]) -> List[int]:
 def compute_reduced_embeddings(embeddings: List[List[float]]) -> List[List[float]]:
     tsne_model = TSNE(n_components=2, random_state=42)
     for emb in embeddings:
+        if len(emb) != 384:
+            print(emb)
         assert len(emb) == 384
     embeddings = np.asarray(embeddings, dtype=object)
     print(embeddings.shape)