使用t-SNE降维,可视化数据的小尝试
from sklearn.manifold import TSNE
import matplotlib.pyplot as plt
tsne = TSNE(n_components=2) #降到2维
X_2D = tsne.fit_transform(X)
plt.scatter(X_2D[:, 0], X_2D[:, 1], c=Y) # c=Y表示按标签Y来填充颜色,Y是label向量
X原来是(55,47,3)的RGB图像,我做了reshape。
可以看出数据分布得太乱,需要做标准化