mlpack
|
LeafSizeNSWrapper wraps any NeighborSearch types that take a leaf size for tree construction. More...
#include <ns_model.hpp>
Public Member Functions | |
LeafSizeNSWrapper (const NeighborSearchMode searchMode, const double epsilon) | |
Construct the LeafSizeNSWrapper by delegating to the NSWrapper constructor. More... | |
virtual | ~LeafSizeNSWrapper () |
Delete the LeafSizeNSWrapper. | |
virtual LeafSizeNSWrapper * | Clone () const |
Return a copy of the LeafSizeNSWrapper. | |
virtual void | Train (arma::mat &&referenceSet, const size_t leafSize, const double, const double) |
Train a model with the given parameters. More... | |
virtual void | Search (arma::mat &&querySet, const size_t k, arma::Mat< size_t > &neighbors, arma::mat &distances, const size_t leafSize, const double) |
Perform bichromatic search (e.g. More... | |
template<typename Archive > | |
void | serialize (Archive &ar, const uint32_t) |
Serialize the NeighborSearch model. | |
![]() | |
NSWrapper (const NeighborSearchMode searchMode, const double epsilon) | |
Construct the NSWrapper object, initializing the internally-held NeighborSearch object. More... | |
virtual | ~NSWrapper () |
Delete the NSWrapper object. | |
const arma::mat & | Dataset () const |
Get a reference to the reference set. | |
NeighborSearchMode | SearchMode () const |
Get the search mode. | |
NeighborSearchMode & | SearchMode () |
Modify the search mode. | |
double | Epsilon () const |
Get epsilon, the approximation parameter. | |
double & | Epsilon () |
Modify epsilon, the approximation parameter. | |
virtual void | Search (const size_t k, arma::Mat< size_t > &neighbors, arma::mat &distances) |
Perform monochromatic neighbor search (i.e. More... | |
template<typename Archive > | |
void | serialize (Archive &ar, const uint32_t) |
Serialize the NeighborSearch model. | |
![]() | |
NSWrapperBase () | |
Create the NSWrapperBase object. More... | |
virtual | ~NSWrapperBase () |
Destruct the NSWrapperBase (nothing to do). | |
Additional Inherited Members | |
![]() | |
typedef NeighborSearch< SortPolicy, metric::EuclideanDistance, arma::mat, TreeType, DualTreeTraversalType, SingleTreeTraversalType > | NSType |
![]() | |
NSType | ns |
The instantiated NeighborSearch object that we are wrapping. | |
LeafSizeNSWrapper wraps any NeighborSearch types that take a leaf size for tree construction.
The implementations of Train() and Search() take the leaf size into account.
|
inline |
Construct the LeafSizeNSWrapper by delegating to the NSWrapper constructor.
|
virtual |
Perform bichromatic search (e.g.
search with a separate query set). This overload uses the leaf size, but ignores the other parameters.
Reimplemented from mlpack::neighbor::NSWrapper< SortPolicy, TreeType, DualTreeTraversalType, SingleTreeTraversalType >.
|
virtual |
Train a model with the given parameters.
This overload uses leafSize but ignores the other parameters.
Reimplemented from mlpack::neighbor::NSWrapper< SortPolicy, TreeType, DualTreeTraversalType, SingleTreeTraversalType >.