[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