X-Git-Url: https://code.communitydata.science/cdsc_reddit.git/blobdiff_plain/ed0e1a82357219ea70e6dfc3396d074af157086f..82d184d9c608db47f5c37f17a18962f07169cbc7:/fit_tsne.py?ds=sidebyside

diff --git a/fit_tsne.py b/fit_tsne.py
index dd9fb93..7de2ac0 100644
--- a/fit_tsne.py
+++ b/fit_tsne.py
@@ -24,7 +24,7 @@ mat = np.stack(mat)
 mat = mat + np.tril(mat.transpose(),k=-1)
 dist = 2*np.arccos(mat)/np.pi
 
-tsne_model = TSNE(2,learning_rate=200,perplexity=40,n_iter=5000,metric='precomputed')
+tsne_model = TSNE(2,learning_rate=750,perplexity=50,n_iter=10000,metric='precomputed',early_exaggeration=20,n_jobs=-1)
 
 tsne_fit_model = tsne_model.fit(dist)