X-Git-Url: https://code.communitydata.science/cdsc_reddit.git/blobdiff_plain/6baa08889b2f46c14f2baa5e3d2136cf165b1673..ed0e1a82357219ea70e6dfc3396d074af157086f:/fit_tsne.py

diff --git a/fit_tsne.py b/fit_tsne.py
index 37341d4..dd9fb93 100644
--- a/fit_tsne.py
+++ b/fit_tsne.py
@@ -28,7 +28,7 @@ tsne_model = TSNE(2,learning_rate=200,perplexity=40,n_iter=5000,metric='precompu
 
 tsne_fit_model = tsne_model.fit(dist)
 
-tsne_fit_whole = tsne_fit_model.fit_transform(mat)
+tsne_fit_whole = tsne_fit_model.fit_transform(dist)
 
 plot_data = pd.DataFrame({'x':tsne_fit_whole[:,0],'y':tsne_fit_whole[:,1], 'subreddit':col_names})