tsne.h File Reference

#include <cuml/common/logger.hpp>
#include <raft/distance/distance_types.hpp>

Include dependency graph for tsne.h:

Go to the source code of this file.

Classes
struct	ML::TSNEParams

Namespaces
	raft
	ML

Enumerations
enum	ML::TSNE_ALGORITHM { ML::EXACT , ML::BARNES_HUT , ML::FFT }
enum	ML::TSNE_INIT { ML::RANDOM , ML::PCA }

Functions
void	ML::TSNE_fit (const raft::handle_t &handle, float *X, float *Y, int n, int p, int64_t *knn_indices, float *knn_dists, TSNEParams &params, float *kl_div=nullptr)
	Dimensionality reduction via TSNE using Barnes-Hut, Fourier Interpolation, or naive methods. or brute force O(N^2). More...
void	ML::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 &params, float *kl_div=nullptr)
	Dimensionality reduction via TSNE using either Barnes Hut O(NlogN) or brute force O(N^2). More...