Skip to content

Commit fbc8125

Browse files
authored
debug changes
1 parent 2eb334d commit fbc8125

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

utils/clusters_utils.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,10 @@ def compute_clusters(embeddings: List[List[float]]) -> List[int]:
1919

2020
def compute_reduced_embeddings(embeddings: List[List[float]]) -> List[List[float]]:
2121
tsne_model = TSNE(n_components=2, random_state=42)
22+
for emb in embeddings:
23+
assert len(emb) == 384
2224
embeddings = np.asarray(embeddings, dtype=object)
25+
print(embeddings.shape)
2326
tsne_data = tsne_model.fit_transform(embeddings)
2427

2528
return (tsne_data - tsne_data.min()).tolist()

0 commit comments

Comments
 (0)