21 #include <cuvs/distance/distance.hpp>
97 rapids_logger::level_enum
verbosity = rapids_logger::level_enum::info;
109 cuvs::distance::DistanceType
metric = cuvs::distance::DistanceType::L2SqrtExpanded;
144 int64_t* knn_indices,
147 float* kl_div =
nullptr);
184 float* kl_div =
nullptr);
Definition: params.hpp:34
Definition: dbscan.hpp:30
void TSNE_fit(const raft::handle_t &handle, float *X, float *Y, int n, int p, int64_t *knn_indices, float *knn_dists, TSNEParams ¶ms, float *kl_div=nullptr)
Dimensionality reduction via TSNE using Barnes-Hut, Fourier Interpolation, or naive methods....
void TSNE_fit_sparse(const raft::handle_t &handle, int *indptr, int *indices, float *data, float *Y, int nnz, int n, int p, int *knn_indices, float *knn_dists, TSNEParams ¶ms, float *kl_div=nullptr)
Dimensionality reduction via TSNE using either Barnes Hut O(NlogN) or brute force O(N^2).
TSNE_ALGORITHM
Definition: tsne.h:29
@ BARNES_HUT
Definition: tsne.h:29
@ FFT
Definition: tsne.h:29
@ EXACT
Definition: tsne.h:29
TSNE_INIT
Definition: tsne.h:31
@ RANDOM
Definition: tsne.h:31
@ PCA
Definition: tsne.h:31
Definition: dbscan.hpp:26
float perplexity
Definition: tsne.h:48
float pre_learning_rate
Definition: tsne.h:73
int perplexity_max_iter
Definition: tsne.h:51
bool square_distances
Definition: tsne.h:106
int exaggeration_iter
Definition: tsne.h:67
float min_grad_norm
Definition: tsne.h:83
cuvs::distance::DistanceType metric
Definition: tsne.h:109
float theta
Definition: tsne.h:42
float late_exaggeration
Definition: tsne.h:62
TSNE_ALGORITHM algorithm
Definition: tsne.h:115
long long random_state
Definition: tsne.h:94
float pre_momentum
Definition: tsne.h:86
int n_neighbors
Definition: tsne.h:38
float early_exaggeration
Definition: tsne.h:58
float post_momentum
Definition: tsne.h:89
int dim
Definition: tsne.h:35
float min_gain
Definition: tsne.h:70
float post_learning_rate
Definition: tsne.h:76
float epssq
Definition: tsne.h:45
TSNE_INIT init
Definition: tsne.h:100
float perplexity_tol
Definition: tsne.h:54
int max_iter
Definition: tsne.h:79
float p
Definition: tsne.h:112
rapids_logger::level_enum verbosity
Definition: tsne.h:97