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

fastlab-svn at coffeetalk-1.cc.gatech.edu fastlab-svn at coffeetalk-1.cc.gatech.edu
Tue Aug 19 11:45:36 EDT 2014


Author: siddharth.950
Date: Tue Aug 19 11:45:36 2014
New Revision: 17075

Log:
Added Reg SVD to CF executable.

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 Aug 19 11:45:36 2014
@@ -8,11 +8,13 @@
 #include <mlpack/core.hpp>
 
 #include <mlpack/methods/amf/amf.hpp>
+#include <mlpack/methods/regularized_svd/regularized_svd.hpp>
 #include "cf.hpp"
 
 using namespace mlpack;
 using namespace mlpack::cf;
 using namespace mlpack::amf;
+using namespace mlpack::svd;
 using namespace std;
 
 // Document program.
@@ -116,7 +118,9 @@
   else if(algo == "SVDIncompleteIncremental") 
     CR(SparseSVDIncompleteIncrementalFactorizer());
   else if(algo == "SVDCompleteIncremental")
-    CR(SparseSVDCompleteIncrementalFactorizer());                 
+    CR(SparseSVDCompleteIncrementalFactorizer());
+  else if(algo == "RegSVD")
+    CR(RegularizedSVD<>());
 
   const string outputFile = CLI::GetParam<string>("output_file");
   data::Save(outputFile, recommendations);



More information about the mlpack-svn mailing list