使用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。

在这里插入图片描述
可以看出数据分布得太乱,需要做标准化

猜你喜欢

转载自blog.csdn.net/qq_37717661/article/details/84977450
今日推荐