pytorch实现t-SNE

pytorch版本实现的t-SNE,可以支持cuda加速,根据作者的python版本修改获得
MNIST数据集在pytorch版本下的结果


pytorch

对比原来python版本实现的结果


python

具体源码,见github
https://github.com/mxl1990/tsne-pytorch

你可能感兴趣的:(pytorch实现t-SNE)