Wrapper function to generate nearest neighbour parameters

Usage

params_nn(
  dist_metric = c("euclidean", "cosine"),
  n_tree = 50L,
  search_budget = NULL,
  m = 16L,
  ef_construction = 100L,
  ef_search = 100L,
  diversify_prob = 0,
  delta = 0.001,
  ef_budget = NULL,
  bt_budget = 0.1
)

Arguments

dist_metric: Character. The distance metric to use. Defaults to "cosine".
n_tree: Integer. Number of trees for Annoy. Defaults to 50L.
search_budget: Integer or NULL. Search budget for Annoy. Defaults to NULL.
m: Integer. Number of bidirectional links for HNSW. Defaults to 16L.
ef_construction: Integer. Size of the dynamic candidate list during HNSW construction. Defaults to 100L.
ef_search: Integer. Size of the dynamic candidate list during HNSW search. Defaults to 100L.
diversify_prob: Float. Diversification probability for NN descent. Defaults to 0.0.
delta: Float. Precision parameter for NN descent. Defaults to 0.001.
ef_budget: Integer or NULL. Effort budget for NN descent. Defaults to NULL.
bt_budget: Float. Budget for ball tree search. Defaults to 0.1.

Value

A list with the nearest neighbour parameters.