[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