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。
可以看出數據分佈得太亂,需要做標準化