diff --git a/utils/clusters_utils.py b/utils/clusters_utils.py index b7270f6..04e9162 100644 --- 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)