umap.hpp File Reference

#include <cuml/manifold/umapparams.h>
#include <raft/core/host_coo_matrix.hpp>
#include <raft/sparse/coo.hpp>
#include <cstddef>
#include <cstdint>
#include <memory>

Include dependency graph for umap.hpp:

Go to the source code of this file.

Namespaces
	raft
	ML
	ML::UMAP

Functions
void	ML::UMAP::find_ab (const raft::handle_t &handle, UMAPParams *params)
std::unique_ptr< raft::sparse::COO< float, int > >	ML::UMAP::get_graph (const raft::handle_t &handle, float *X, float *y, int n, int d, int64_t *knn_indices, float *knn_dists, UMAPParams *params)
void	ML::UMAP::refine (const raft::handle_t &handle, float *X, int n, int d, raft::sparse::COO< float, int > *graph, UMAPParams *params, float *embeddings)
void	ML::UMAP::init_and_refine (const raft::handle_t &handle, float *X, int n, int d, raft::sparse::COO< float, int > *graph, UMAPParams *params, float *embeddings)
void	ML::UMAP::fit (const raft::handle_t &handle, float *X, float *y, int n, int d, int64_t *knn_indices, float *knn_dists, UMAPParams *params, float *embeddings, raft::host_coo_matrix< float, int, int, uint64_t > &graph)
void	ML::UMAP::fit_sparse (const raft::handle_t &handle, int *indptr, int *indices, float *data, size_t nnz, float *y, int n, int d, int *knn_indices, float *knn_dists, UMAPParams *params, float *embeddings, raft::host_coo_matrix< float, int, int, uint64_t > &graph)
void	ML::UMAP::transform (const raft::handle_t &handle, float *X, int n, int d, float *orig_X, int orig_n, float *embedding, int embedding_n, UMAPParams *params, float *transformed)
void	ML::UMAP::transform_sparse (const raft::handle_t &handle, int *indptr, int *indices, float *data, size_t nnz, int n, int d, int *orig_x_indptr, int *orig_x_indices, float *orig_x_data, size_t orig_nnz, int orig_n, float *embedding, int embedding_n, UMAPParams *params, float *transformed)