#include <common/cumlHandle.hpp>
#include <cuml/neighbors/knn.hpp>
#include <cuml/neighbors/knn_api.h>
#include <vector>
Namespaces | |
ML | |
Functions | |
cumlError_t | ML::knn_search (const cumlHandle_t handle, float **input, int *sizes, int n_params, int D, float *search_items, int n, int64_t *res_I, float *res_D, int k, bool rowMajorIndex, bool rowMajorQuery, int metric_type, float metric_arg, bool expanded) |
Flat C API function to perform a brute force knn on a series of input arrays and combine the results into a single output array for indexes and distances. More... | |