We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2eb334d commit fbc8125Copy full SHA for fbc8125
utils/clusters_utils.py
@@ -19,7 +19,10 @@ def compute_clusters(embeddings: List[List[float]]) -> List[int]:
19
20
def compute_reduced_embeddings(embeddings: List[List[float]]) -> List[List[float]]:
21
tsne_model = TSNE(n_components=2, random_state=42)
22
+ for emb in embeddings:
23
+ assert len(emb) == 384
24
embeddings = np.asarray(embeddings, dtype=object)
25
+ print(embeddings.shape)
26
tsne_data = tsne_model.fit_transform(embeddings)
27
28
return (tsne_data - tsne_data.min()).tolist()
0 commit comments