[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