Public Attributes | List of all members
ML::knnIndex Struct Reference

#include <knn.hpp>

Collaboration diagram for ML::knnIndex:
Collaboration graph

Public Attributes

raft::distance::DistanceType metric
 
float metricArg
 
int nprobe
 
std::unique_ptr< raft::spatial::knn::MetricProcessor< float > > metric_processor
 
std::unique_ptr< cuvs::neighbors::ivf_flat::index< float, int64_t > > ivf_flat
 
std::unique_ptr< cuvs::neighbors::ivf_pq::index< int64_t > > ivf_pq
 
int device
 

Member Data Documentation

◆ device

int ML::knnIndex::device

◆ ivf_flat

std::unique_ptr<cuvs::neighbors::ivf_flat::index<float, int64_t> > ML::knnIndex::ivf_flat

◆ ivf_pq

std::unique_ptr<cuvs::neighbors::ivf_pq::index<int64_t> > ML::knnIndex::ivf_pq

◆ metric

raft::distance::DistanceType ML::knnIndex::metric

◆ metric_processor

std::unique_ptr<raft::spatial::knn::MetricProcessor<float> > ML::knnIndex::metric_processor

◆ metricArg

float ML::knnIndex::metricArg

◆ nprobe

int ML::knnIndex::nprobe

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