small refactoring and kicked the visualizer again

2019-12-15 09:51:22 -08:00
parent 06d0b2d565
commit 05480606b0
4 changed files with 82 additions and 30 deletions
--- a/library.py
+++ b/library.py
@@ -113,6 +113,13 @@ def get_embeddings(net):
    return net.get_weights()[0]


+def calc_TSNE(emb):
+    # import umap
+    # return umap.UMAP().fit_transform(emb)
+    return emb
+
+
+
 def save_embeddings(emb):
    import numpy as np
    np.savetxt(os.path.join(RESULTS, f'embeddings_{CFG["data_name"]}.csv'),