Merge pull request #58 from rjavierch/bugfix/issue-54

Update tsne function due to dependency change in nlp.py
NLeSC · Aug 28, 2023 · ab9f1d6 · ab9f1d6
2 parents 242154d + 85ddb2c
commit ab9f1d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/litstudy/nlp.py b/litstudy/nlp.py
@@ -408,6 +408,6 @@ def calculate_embedding(corpus: Corpus, *, rank=2, svd_dims=50, perplexity=30, s
         components = tfidf
 
     model = TSNE(
-        rank, metric="cosine", square_distances=True, perplexity=perplexity, random_state=seed
+        rank, metric="cosine", perplexity=perplexity, random_state=seed
     )
     return model.fit_transform(components)