Public Attributes | List of all members
ML::TSNEParams Struct Reference

#include <tsne.h>

Collaboration diagram for ML::TSNEParams:
Collaboration graph

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
 
cuvs::distance::DistanceType metric = cuvs::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

cuvs::distance::DistanceType ML::TSNEParams::metric = cuvs::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: