写一个画MNIST数据集的TSNE图的程序
时间: 2023-02-19 16:39:05 浏览: 81
可以使用scikit-learn库中的TSNE类来实现这个功能,示例代码如下:
from sklearn.manifold import TSNE
import numpy as np
# 加载MNIST数据集
mnist_data = np.load('mnist.npy')
# 将MNIST数据集转换为TSNE图
tsne = TSNE(n_components=2)
tsne_result = tsne.fit_transform(mnist_data)
# 绘制TSNE图
plt.scatter(tsne_result[:, 0], tsne_result[:, 1])
plt.show()
阅读全文