org.deeplearning4j.plot

Class Tsne

java.lang.Object

org.deeplearning4j.plot.Tsne

Direct Known Subclasses:

BarnesHutTsne

public class Tsne
extends Object

dl4j port of original t-sne algorithm described/implemented by van der Maaten and Hinton DECOMPOSED VERSION, DO NOT USE IT EVER

Author:

[email protected], Adam Gibson

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	Tsne.Builder

Field Summary

Fields

Modifier and Type	Field and Description
protected org.nd4j.linalg.learning.AdaGrad	adaGrad
protected double	finalMomentum
protected double	initialMomentum
protected IterationListener	iterationListener
protected double	learningRate
protected static org.slf4j.Logger	logger
protected int	maxIter
protected double	minGain
protected double	momentum
protected boolean	normalize
protected double	perplexity
protected double	realMin
protected int	stopLyingIteration
protected int	switchMomentumIteration
protected double	tolerance
protected boolean	useAdaGrad
protected boolean	usePca
protected org.nd4j.linalg.api.ndarray.INDArray	Y

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	Tsne()

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.api.ndarray.INDArray	calculate(org.nd4j.linalg.api.ndarray.INDArray X, int targetDimensions, double perplexity)
org.nd4j.linalg.api.ndarray.INDArray	diag(org.nd4j.linalg.api.ndarray.INDArray ds)
Pair<Double,org.nd4j.linalg.api.ndarray.INDArray>	hBeta(org.nd4j.linalg.api.ndarray.INDArray d, double beta) Computes a gaussian kernel given a vector of squared distance distances
protected void	init()
void	plot(org.nd4j.linalg.api.ndarray.INDArray matrix, int nDims, List<String> labels, String path)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

maxIter

protected int maxIter

realMin

protected double realMin

initialMomentum

protected double initialMomentum

finalMomentum

protected double finalMomentum

minGain

protected double minGain

momentum

protected double momentum

switchMomentumIteration

protected int switchMomentumIteration

normalize

protected boolean normalize

usePca

protected boolean usePca

stopLyingIteration

protected int stopLyingIteration

tolerance

protected double tolerance

learningRate

protected double learningRate

adaGrad

protected org.nd4j.linalg.learning.AdaGrad adaGrad

useAdaGrad

protected boolean useAdaGrad

perplexity

protected double perplexity

Y

protected org.nd4j.linalg.api.ndarray.INDArray Y

iterationListener

protected transient IterationListener iterationListener

logger

protected static final org.slf4j.Logger logger

Constructor Detail

Tsne

protected Tsne()

Method Detail

init

protected void init()

calculate

public org.nd4j.linalg.api.ndarray.INDArray calculate(org.nd4j.linalg.api.ndarray.INDArray X,
                                                      int targetDimensions,
                                                      double perplexity)

diag

public org.nd4j.linalg.api.ndarray.INDArray diag(org.nd4j.linalg.api.ndarray.INDArray ds)

plot

public void plot(org.nd4j.linalg.api.ndarray.INDArray matrix,
                 int nDims,
                 List<String> labels,
                 String path)
          throws IOException

Throws:

IOException

hBeta

public Pair<Double,org.nd4j.linalg.api.ndarray.INDArray> hBeta(org.nd4j.linalg.api.ndarray.INDArray d,
                                                               double beta)

Computes a gaussian kernel given a vector of squared distance distances

Parameters:

d - the data

beta -

Returns: