tsne降维可视化

import matplotlib.pyplot as plt
from sklearn.manifold import TSNE

tsne = TSNE()
tsne.fit_transform(data_scale)

tsne = pd.DataFrame(tsne.embedding_,index = data_scale.index)#tsne.embedding_即降维后的二维数据

d = tsne[data['labels']==0]
plt.plot(d[0],d[1],'r.')
d = tsne[data['labels']==1]
plt.plot(d[0],d[1],'go')
d = tsne[data['labels']==2]
plt.plot(d[0],d[1],'b*')

原来三维的数据转为二维可视化。