T-SNE+Python散点图绘制+图例

T-SNE+Python散点图绘制+图例

背景

T-SNE可以用于数据降维，降维之后的数据我们用散点图进行可视化处理。

在这里我搜集了不少资料，发现散点图添加图例的最好方法还是把不同类的点分开来存储，在绘制的时候也分开来绘制。

具体代码见下，已有十分详细的注释。

代码

import matplotlib.pyplot as plt
from sklearn import manifold, datasets
import scipy.io as scio

# x中保存的是数值矩阵， 每个点的特征为一行

dataFile = 'F:\Image-steganography\\tsne\Data_test.mat'
x = scio.loadmat(dataFile)
x = x['Data_test']

# color是为了可视化做的颜色
dataFile = 'F:\Image-steganography\\tsne\label_test.mat'
color = scio.loadmat