[mlpack-svn] r15900 - mlpack/trunk/src/mlpack/methods/cf

fastlab-svn at coffeetalk-1.cc.gatech.edu fastlab-svn at coffeetalk-1.cc.gatech.edu
Tue Oct 1 16:18:33 EDT 2013


Author: rcurtin
Date: Tue Oct  1 16:18:33 2013
New Revision: 15900

Log:
Actually use number of recommendations and size of neighborhood in calculations.


Modified:
   mlpack/trunk/src/mlpack/methods/cf/cf_main.cpp

Modified: mlpack/trunk/src/mlpack/methods/cf/cf_main.cpp
==============================================================================
--- mlpack/trunk/src/mlpack/methods/cf/cf_main.cpp	(original)
+++ mlpack/trunk/src/mlpack/methods/cf/cf_main.cpp	Tue Oct  1 16:18:33 2013
@@ -37,10 +37,10 @@
 //PARAM_STRING("nearest_neighbor_algorithm", "Similarity search procedure to "
 //    "be used for generating recommendations.", "s", "knn");
 
-PARAM_INT("number_of_Recommendations", "Number of Recommendations for each "
-          "user in query", "r",5);
-PARAM_INT("neighbourhood", "Size of the neighbourhood for all "
-          "user in query", "n",5);
+PARAM_INT("recommendations", "Number of recommendations to generate for each "
+    "query user.", "r", 5);
+PARAM_INT("neighborhood", "Size of the neighborhood of similar users to "
+    "consider for each query user.", "n", 5);
 
 int main(int argc, char** argv)
 {
@@ -55,9 +55,15 @@
   // Recommendation matrix.
   arma::Mat<size_t> recommendations;
 
+  // Get parameters.
+  const size_t numRecs = (size_t) CLI::GetParam<int>("recommendations");
+  const size_t neighborhood = (size_t) CLI::GetParam<int>("neighborhood");
+
   // Perform decomposition to prepare for recommendations.
   Log::Info << "Performing CF matrix decomposition on dataset..." << endl;
   CF c(dataset);
+  c.NumRecs(numRecs);
+  c.NumUsersForSimilarity(neighborhood);
 
   // Reading users.
   const string queryFile = CLI::GetParam<string>("query_file");



More information about the mlpack-svn mailing list