12 #ifndef MLPACK_METHODS_FASTMKS_FASTMKS_MODEL_HPP
13 #define MLPACK_METHODS_FASTMKS_FASTMKS_MODEL_HPP
63 template<
typename TKernelType>
64 void BuildModel(
const arma::mat& referenceData,
66 const bool singleMode,
97 void Search(
const arma::mat& querySet,
99 arma::Mat<size_t>& indices,
112 void Search(
const size_t k,
113 arma::Mat<size_t>& indices,
119 template<
typename Archive>
120 void Serialize(Archive& ar,
const unsigned int );
142 template<
typename FastMKSType>
143 void Search(FastMKSType& f,
144 const arma::mat& querySet,
146 arma::Mat<size_t>& indices,
154 #include "fastmks_model_impl.hpp"
int KernelType() const
Get the kernel type.
The core includes that mlpack expects; standard C++ includes and Armadillo.
bool SingleMode() const
Get whether or not single-tree search is used.
void Serialize(Archive &ar, const unsigned int)
Serialize the model.
~FastMKSModel()
Clean memory.
bool Naive() const
Get whether or not naive search is used.
A utility struct to contain all the possible FastMKS models, for use by the mlpack_fastmks program...
void BuildModel(const arma::mat &referenceData, TKernelType &kernel, const bool singleMode, const bool naive, const double base)
Build the model on the given reference set.
void Search(const arma::mat &querySet, const size_t k, arma::Mat< size_t > &indices, arma::mat &kernels, const double base)
Search with a different query set.
FastMKSModel(const int kernelType=LINEAR_KERNEL)
Create the FastMKSModel with the given kernel type.
KernelTypes
A list of all the kernels we support.
int & KernelType()
Modify the kernel type.