tsne python_Python / Tensorflow / Keras实现参数tSNE算法

Overview

This is a python package implementing parametric t-SNE. We train a neural-network to learn a mapping by minimizing the Kullback-Leibler divergence between the Gaussian distance metric in the high-dimensional space and the Students-t distributed distance metric in the low-dimensional space. By default we use similar archictecture1 as van der Maaten 2009, which is a dense neural network with layers: [input dimension], 500, 500, 2000, [output dimension]

Simple example usage may be:

train_data = load_training_data_somehow()

high_dims = train_data.shape[1]

num_outputs = 2

perplexity = 30

ptSNE = Parametric_tSNE(high_dims, num_outputs, perplexity)

ptSNE.fit(train_data)

output_res = ptSNE.transform(train_data)

output_res will be N x num_outputs, the transformation of each point. At this point, ptSNE will be a trained model, so we can qui