绘制t-SNE图

什么是t-SNE图？

如下图，下图来源于论文Contrastive Clustering
一般用于分类问题/对比学习。

作用？

体现出经过层层训练，类内越来越紧密，类间差异越来越大；或者也可以做消融可视化。

怎么画？

以下是一个手写数字的例子，转载自添加链接描述

import numpy as np
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt_sne
from sklearn import datasets
from sklearn.manifold import TSNE
import os
 
 
def plot_tsne(features, labels, epoch,fileNameDir = None):
    '''
    features:(N*m) N*m大小特征，其中N代表有N个数据，每个数据m维
    label:(N) 有N个标签
    '''
    print(features.shape,labels.shape)
    print(type(features),type(labels))
    print(np.any(np.isnan(features)),np.any(np.isinf(features)))
    features = np.nan_to_num