t-SNE算法可视化数据和卷积层

一、可视化数据

t-SNE算法是一种用于数据降维和可视化的非线性方法。在一维数据的情况下,t-SNE算法可以用于可视化数据点之间的相似度关系,以下是一个简单的示例代码:

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

# 生成一维数据
data = np.array([1, 3, 5, 7, 8, 10, 12, 13, 14, 15, 17, 19, 20, 22, 24, 25, 27, 28, 30, 32])
label = np.array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1])

# 使用t-SNE算法进行降维
tsne = TSNE(n_components=1, random_state=0)
embedding = tsne.fit_transform(data.reshape(-1, 1))

# 可视化数据
plt.scatter(embedding[:, 0], np.zeros_like(embedding[:, 0]), c=label)
plt.show()

在这个示例中,我们生成了一维数据,其中前10个数据点属于类别0,后10个数据点属于类别1。然后,我们使用t-SNE算法将数据降维到一维,最后将降维后的数据点可视化。可以看到,数据点已经被分成了两个明显的类别,并且它们在一维空间中的位置已经反映了它们在原始数据中的相似度关系。
在这里插入图片描述

二、可视化卷积层

以下是一个简单的示例代码,使用pytorch实现t-SNE算法可视化卷积层的输出特征:

import torch
import torch.nn as nn
import torchvision
from sklearn.manifold import TSNE
import matplotlib.pyplot as plt

# 加载预训练的VGG16模型
model = torchvision.models.vgg16(pretrained=False)

# 获取卷积层的输出特征
features = nn.Sequential(*list(model.features.children())).eval()
conv_outputs = []
def hook(module, input, output):
    conv_outputs.append(output)
features[0].register_forward_hook(hook)
data = torch.randn(10, 3, 224, 224)
output = features(data)

# 使用t-SNE算法降维
tsne = TSNE(n_components=2, verbose=1, perplexity=30, n_iter=1000)
conv_outputs = torch.cat(conv_outputs)
conv_outputs = conv_outputs.view(conv_outputs.size(0), -1)
tsne_results = tsne.fit_transform(conv_outputs.detach().numpy())

# 绘制可视化图像
plt.scatter(tsne_results[:, 0], tsne_results[:, 1])
plt.show()

在上述代码中,我们使用了预训练的VGG16模型,并获取了第一个卷积层的输出特征。然后使用t-SNE算法将其降维为二维,最后将结果可视化。需要注意的是,在降维时需要对输出特征进行展平操作。

猜你喜欢

转载自blog.csdn.net/qq_23345187/article/details/129358342