I\'m currently using t-SNE to visualize data features. First, I extracted the features vector from my model.
import torch.nn as nn class Net(nn.Module):