mlpack  2.2.5
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
Public Types | Public Member Functions | List of all members
FastMKSModel Class Reference

A utility struct to contain all the possible FastMKS models, for use by the mlpack_fastmks program. More...

Public Types

enum  KernelTypes {
  LINEAR_KERNEL,
  POLYNOMIAL_KERNEL,
  COSINE_DISTANCE,
  GAUSSIAN_KERNEL,
  EPANECHNIKOV_KERNEL,
  TRIANGULAR_KERNEL,
  HYPTAN_KERNEL
}
 A list of all the kernels we support. More...
 

Public Member Functions

 FastMKSModel (const int kernelType=LINEAR_KERNEL)
 Create the FastMKSModel with the given kernel type. More...
 
 ~FastMKSModel ()
 Clean memory. More...
 
template<typename TKernelType >
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. More...
 
int KernelType () const
 Get the kernel type. More...
 
int & KernelType ()
 Modify the kernel type. More...
 
bool Naive () const
 Get whether or not naive search is used. More...
 
bool & Naive ()
 Set whether or not naive search is used. More...
 
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. More...
 
void Search (const size_t k, arma::Mat< size_t > &indices, arma::mat &kernels)
 Search with the reference set as the query set. More...
 
template<typename Archive >
void Serialize (Archive &ar, const unsigned int)
 Serialize the model. More...
 
bool SingleMode () const
 Get whether or not single-tree search is used. More...
 
bool & SingleMode ()
 Set whether or not single-tree search is used. More...
 

Detailed Description

A utility struct to contain all the possible FastMKS models, for use by the mlpack_fastmks program.

Definition at line 34 of file fastmks_model.hpp.

Member Enumeration Documentation

A list of all the kernels we support.

Enumerator
LINEAR_KERNEL 
POLYNOMIAL_KERNEL 
COSINE_DISTANCE 
GAUSSIAN_KERNEL 
EPANECHNIKOV_KERNEL 
TRIANGULAR_KERNEL 
HYPTAN_KERNEL 

Definition at line 38 of file fastmks_model.hpp.

Constructor & Destructor Documentation

FastMKSModel ( const int  kernelType = LINEAR_KERNEL)

Create the FastMKSModel with the given kernel type.

Clean memory.

Member Function Documentation

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.

Make sure kernelType is equal to the correct entry in KernelTypes for the given KernelType class!

int KernelType ( ) const
inline

Get the kernel type.

Definition at line 81 of file fastmks_model.hpp.

int& KernelType ( )
inline

Modify the kernel type.

Definition at line 83 of file fastmks_model.hpp.

bool Naive ( ) const

Get whether or not naive search is used.

bool& Naive ( )

Set whether or not naive search is used.

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.

Parameters
querySetSet to search with.
kNumber of max-kernel candidates to search for.
indicesA matrix in which to store the indices of max-kernel candidates.
kernelsA matrix in which to store the max-kernel candidate kernel values.
baseBase to use for cover tree building (if in dual-tree search mode).
void Search ( const size_t  k,
arma::Mat< size_t > &  indices,
arma::mat &  kernels 
)

Search with the reference set as the query set.

Parameters
kNumber of max-kernel candidates to search for.
indicesA matrix in which to store the indices of max-kernel candidates.
kernelsA matrix in which to store the max-kernel candidate kernel values.
void Serialize ( Archive &  ar,
const unsigned  int 
)

Serialize the model.

bool SingleMode ( ) const

Get whether or not single-tree search is used.

bool& SingleMode ( )

Set whether or not single-tree search is used.


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