ML::TSNEParams Struct Reference

#include <tsne.h>

Collaboration diagram for ML::TSNEParams:

Public Attributes
int	dim = 2
int	n_neighbors = 1023
float	theta = 0.5f
float	epssq = 0.0025
float	perplexity = 50.0f
int	perplexity_max_iter = 100
float	perplexity_tol = 1e-5
float	early_exaggeration = 12.0f
float	late_exaggeration = 1.0f
int	exaggeration_iter = 250
float	min_gain = 0.01f
float	pre_learning_rate = 200.0f
float	post_learning_rate = 500.0f
int	max_iter = 1000
float	min_grad_norm = 1e-7
float	pre_momentum = 0.5
float	post_momentum = 0.8
long long	random_state = -1
int	verbosity = CUML_LEVEL_INFO
TSNE_INIT	init = TSNE_INIT::RANDOM
bool	square_distances = true
raft::distance::DistanceType	metric = raft::distance::DistanceType::L2SqrtExpanded
float	p = 2.0
TSNE_ALGORITHM	algorithm = TSNE_ALGORITHM::FFT

Member Data Documentation

algorithm

TSNE_ALGORITHM ML::TSNEParams::algorithm = TSNE_ALGORITHM::FFT

dim

int ML::TSNEParams::dim = 2

early_exaggeration

float ML::TSNEParams::early_exaggeration = 12.0f

epssq

float ML::TSNEParams::epssq = 0.0025

exaggeration_iter

int ML::TSNEParams::exaggeration_iter = 250

init

TSNE_INIT ML::TSNEParams::init = TSNE_INIT::RANDOM

late_exaggeration

float ML::TSNEParams::late_exaggeration = 1.0f

max_iter

int ML::TSNEParams::max_iter = 1000

metric

raft::distance::DistanceType ML::TSNEParams::metric = raft::distance::DistanceType::L2SqrtExpanded

min_gain

float ML::TSNEParams::min_gain = 0.01f

min_grad_norm

float ML::TSNEParams::min_grad_norm = 1e-7

n_neighbors

int ML::TSNEParams::n_neighbors = 1023

p

float ML::TSNEParams::p = 2.0

perplexity

float ML::TSNEParams::perplexity = 50.0f

perplexity_max_iter

int ML::TSNEParams::perplexity_max_iter = 100

perplexity_tol

float ML::TSNEParams::perplexity_tol = 1e-5

post_learning_rate

float ML::TSNEParams::post_learning_rate = 500.0f

post_momentum

float ML::TSNEParams::post_momentum = 0.8

pre_learning_rate

float ML::TSNEParams::pre_learning_rate = 200.0f

pre_momentum

float ML::TSNEParams::pre_momentum = 0.5

random_state

long long ML::TSNEParams::random_state = -1

square_distances

bool ML::TSNEParams::square_distances = true

theta

float ML::TSNEParams::theta = 0.5f

verbosity

int ML::TSNEParams::verbosity = CUML_LEVEL_INFO

---

The documentation for this struct was generated from the following file:

• include/cuml/manifold/tsne.h