ver 0.15.11

Author: AnonymousCKNN

npy/tools/tsne.py

@@ -23,13 +23,15 @@ def tsne_multirank(X, **kwargs):
     return Y
 
 
-def draw_tsne(X, n=4):  # [C, N, D]
+def draw_tsne(X, n=4, **kwargs):  # [C, N, D]
     emb = tsne_multirank(X, n_jobs=n)
     C = len(emb)
 
     for c in range(C):
         print(c, len(emb[c]))
-        plt.scatter(*emb[c].T, s=10)
+
+        _kwargs = {k: v[c] if isinstance(v, list) else v for k, v in kwargs.items()}
+        plt.scatter(*emb[c].T, s=10, **_kwargs)
 
 
 def test():

setup.py

@@ -10,7 +10,7 @@
 
 setup(name='nuclear-python',
 
-      version='0.15.10',
+      version='0.15.11',
 
       url='https://github.com/nuclearboy95/nuclear-python',