[mlpack-git] master: Use NeighborSearchMode enum type in cf. (844ea62)
gitdub at mlpack.org
gitdub at mlpack.org
Sat Aug 20 18:28:15 EDT 2016
Repository : https://github.com/mlpack/mlpack
On branch : master
Link : https://github.com/mlpack/mlpack/compare/3274b05fcc545c3b36f783316fea2e22f79c3d03...1c77230c7d3b9c45fb102cd3c632d9c7248e085e
>---------------------------------------------------------------
commit 844ea625bb91d6b6166beef5e4958c5c1dae31dc
Author: MarcosPividori <marcos.pividori at gmail.com>
Date: Sat Aug 20 17:46:32 2016 -0300
Use NeighborSearchMode enum type in cf.
>---------------------------------------------------------------
844ea625bb91d6b6166beef5e4958c5c1dae31dc
src/mlpack/methods/cf/cf.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/mlpack/methods/cf/cf.cpp b/src/mlpack/methods/cf/cf.cpp
index 82e08c1..512d01a 100644
--- a/src/mlpack/methods/cf/cf.cpp
+++ b/src/mlpack/methods/cf/cf.cpp
@@ -159,7 +159,7 @@ double CF::Predict(const size_t user, const size_t item) const
// Calculate the neighborhood of the queried users.
// This should be a templatized option.
- neighbor::KNN a(stretchedH, false, true /* single-tree mode */);
+ neighbor::KNN a(stretchedH, neighbor::SINGLE_TREE_MODE);
arma::mat resultingDistances; // Temporary storage.
a.Search(query, numUsersForSimilarity, neighborhood, resultingDistances);
More information about the mlpack-git
mailing list