knn_mg.hpp File Reference

#include <cumlprims/opg/matrix/data.hpp>
#include <cumlprims/opg/matrix/part_descriptor.hpp>
#include <raft/core/handle.hpp>
#include <vector>

Include dependency graph for knn_mg.hpp:

Go to the source code of this file.

Namespaces
	ML
	ML::KNN
	ML::KNN::opg

Functions
void	ML::KNN::opg::knn (raft::handle_t &handle, std::vector< Matrix::Data< int64_t > * > *out_I, std::vector< Matrix::floatData_t * > *out_D, std::vector< Matrix::floatData_t * > &idx_data, Matrix::PartDescriptor &idx_desc, std::vector< Matrix::floatData_t * > &query_data, Matrix::PartDescriptor &query_desc, bool rowMajorIndex, bool rowMajorQuery, int k, size_t batch_size, bool verbose)
void	ML::KNN::opg::knn_classify (raft::handle_t &handle, std::vector< Matrix::Data< int > * > *out, std::vector< std::vector< float * >> *probas, std::vector< Matrix::floatData_t * > &idx_data, Matrix::PartDescriptor &idx_desc, std::vector< Matrix::floatData_t * > &query_data, Matrix::PartDescriptor &query_desc, std::vector< std::vector< int * >> &y, std::vector< int * > &uniq_labels, std::vector< int > &n_unique, bool rowMajorIndex=false, bool rowMajorQuery=false, bool probas_only=false, int k=10, size_t batch_size=1<< 15, bool verbose=false)
void	ML::KNN::opg::knn_regress (raft::handle_t &handle, std::vector< Matrix::Data< float > * > *out, std::vector< Matrix::floatData_t * > &idx_data, Matrix::PartDescriptor &idx_desc, std::vector< Matrix::floatData_t * > &query_data, Matrix::PartDescriptor &query_desc, std::vector< std::vector< float * >> &y, bool rowMajorIndex, bool rowMajorQuery, int k, int n_outputs, size_t batch_size, bool verbose)