Fix bugs; add KernelType().

This commit is contained in:
ryan
2015-12-22 23:25:57 -05:00
parent fbc045abcd
commit 42a94dbf97
2 changed files with 9 additions and 4 deletions
+7 -2
View File
@@ -44,9 +44,9 @@ class FastMKSModel
* Build the model on the given reference set. Make sure kernelType is equal
* to the correct entry in KernelTypes for the given KernelType class!
*/
template<typename KernelType>
template<typename TKernelType>
void BuildModel(const arma::mat& referenceData,
KernelType& kernel,
TKernelType& kernel,
const bool singleMode,
const bool naive,
const double base);
@@ -61,6 +61,11 @@ class FastMKSModel
//! Set whether or not single-tree search is used.
bool& SingleMode();
//! Get the kernel type.
int KernelType() const { return kernelType; }
//! Modify the kernel type.
int& KernelType() { return kernelType; }
/**
* Search with a different query set.
*
@@ -49,9 +49,9 @@ void BuildFastMKSModel(FastMKSType& /* f */,
" not equal to kernel type of the model!");
}
template<typename KernelType>
template<typename TKernelType>
void FastMKSModel::BuildModel(const arma::mat& referenceData,
KernelType& kernel,
TKernelType& kernel,
const bool singleMode,
const bool naive,
const double base)