Namespaces | Functions
knn_api.cpp File Reference
#include <common/cumlHandle.hpp>
#include <cuml/neighbors/knn.hpp>
#include <cuml/neighbors/knn_api.h>
#include <vector>
Include dependency graph for knn_api.cpp:

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...