print embeddings

ARBML · Apr 6, 2024 · 03bdaa7 · 03bdaa7
1 parent 8f2f624
commit 03bdaa7
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/utils/clusters_utils.py b/utils/clusters_utils.py
@@ -20,9 +20,7 @@ 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
+        print(len(emb))
     embeddings = np.asarray(embeddings, dtype=object)
     print(embeddings.shape)
     tsne_data = tsne_model.fit_transform(embeddings)