ML::graph_build_params::nn_descent_params_umap Struct Reference

#include <umapparams.h>

Collaboration diagram for ML::graph_build_params::nn_descent_params_umap:

Public Attributes
size_t	graph_degree = 64
size_t	intermediate_graph_degree = 128
size_t	max_iterations = 20
float	termination_threshold = 0.0001

Detailed Description

Arguments for using nn descent as the knn build algorithm. graph_degree must be larger than or equal to n_neighbors. Increasing graph_degree and max_iterations may result in better accuracy. Smaller termination threshold means stricter convergence criteria for nn descent and may take longer to converge.

Member Data Documentation

graph_degree

size_t ML::graph_build_params::nn_descent_params_umap::graph_degree = 64

intermediate_graph_degree

size_t ML::graph_build_params::nn_descent_params_umap::intermediate_graph_degree = 128

max_iterations

size_t ML::graph_build_params::nn_descent_params_umap::max_iterations = 20

termination_threshold

float ML::graph_build_params::nn_descent_params_umap::termination_threshold = 0.0001

---

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

• include/cuml/manifold/umapparams.h