用TSNE实现特征降维可视化

背景

用于深度学习里面的多分类,希望直观地观察到类间距离,从而确定具体哪些类别的识别准确率高.

tSNE效果

代码实现

import time
import numpy as np
import matplotlib.pyplot as plt
from sklearn import (manifold, datasets, decomposition, ensemble, discriminant_analysis, random_projection)
import torch
import torch.nn.functional as F


def plot_embedding(X, y, title=None):
    x_min, x_max = np.min(X, 0), np.max(X, 0)
    X = (X - x_min) / (x_max - x_min)

    plt.figure()
    # ax = plt.subplot(111)
    for i in range(X.shape[0]):
        plt.text(X[i