-
- Downloads
FIX t-SNE memory usage and many other optimizer issues (#9032)
Use a sparse matrix representation of the neighbors. Re-factored the QuadTree implementation to avoid insertion errors. Various fixes in the gradient descent schedule to get the Barnes Hut and exact solvers to behave more robustly and consistently.
Showing
- benchmarks/.gitignore 4 additions, 0 deletionsbenchmarks/.gitignore
- benchmarks/bench_tsne_mnist.py 169 additions, 0 deletionsbenchmarks/bench_tsne_mnist.py
- benchmarks/plot_tsne_mnist.py 30 additions, 0 deletionsbenchmarks/plot_tsne_mnist.py
- doc/whats_new.rst 17 additions, 0 deletionsdoc/whats_new.rst
- examples/manifold/plot_t_sne_perplexity.py 38 additions, 6 deletionsexamples/manifold/plot_t_sne_perplexity.py
- sklearn/manifold/_barnes_hut_tsne.pyx 110 additions, 695 deletionssklearn/manifold/_barnes_hut_tsne.pyx
- sklearn/manifold/_utils.pyx 28 additions, 40 deletionssklearn/manifold/_utils.pyx
- sklearn/manifold/setup.py 1 addition, 0 deletionssklearn/manifold/setup.py
- sklearn/manifold/t_sne.py 229 additions, 265 deletionssklearn/manifold/t_sne.py
- sklearn/manifold/tests/test_t_sne.py 234 additions, 124 deletionssklearn/manifold/tests/test_t_sne.py
- sklearn/mixture/base.py 1 addition, 1 deletionsklearn/mixture/base.py
- sklearn/neighbors/quad_tree.pxd 100 additions, 0 deletionssklearn/neighbors/quad_tree.pxd
- sklearn/neighbors/quad_tree.pyx 672 additions, 0 deletionssklearn/neighbors/quad_tree.pyx
- sklearn/neighbors/setup.py 4 additions, 0 deletionssklearn/neighbors/setup.py
- sklearn/neighbors/tests/test_quad_tree.py 108 additions, 0 deletionssklearn/neighbors/tests/test_quad_tree.py
- sklearn/tree/_utils.pxd 3 additions, 1 deletionsklearn/tree/_utils.pxd
Loading
Please register or sign in to comment