[mlpack-svn] r13329 - mlpack/trunk/src/mlpack/methods/local_coordinate_coding

[fastlab-svn at coffeetalk-1.cc.gatech.edu]

Author: rcurtin
Date: 2012-08-03 15:58:59 -0400 (Fri, 03 Aug 2012)
New Revision: 13329

Modified:
   mlpack/trunk/src/mlpack/methods/local_coordinate_coding/lcc_main.cpp
Log:
Allow objective tolerance option.


Modified: mlpack/trunk/src/mlpack/methods/local_coordinate_coding/lcc_main.cpp
===================================================================
--- mlpack/trunk/src/mlpack/methods/local_coordinate_coding/lcc_main.cpp	2012-08-03 19:50:15 UTC (rev 13328)
+++ mlpack/trunk/src/mlpack/methods/local_coordinate_coding/lcc_main.cpp	2012-08-03 19:58:59 UTC (rev 13329)
@@ -60,6 +60,9 @@
 
 PARAM_INT("seed", "Random seed.  If 0, 'std::time(NULL)' is used.", "s", 0);
 
+PARAM_DOUBLE("objective_tolerance", "Tolerance for objective function.", "o",
+    0.01);
+
 using namespace arma;
 using namespace std;
 using namespace mlpack;
@@ -89,6 +92,8 @@
 
   const bool normalize = CLI::HasParam("normalize");
 
+  const double objTolerance = CLI::GetParam<double>("objective_tolerance");
+
   mat input;
   data::Load(inputFile, input, true);
 
@@ -127,7 +132,7 @@
     }
 
     // Run LCC.
-    lcc.Encode(maxIterations);
+    lcc.Encode(maxIterations, objTolerance);
 
     // Save the results.
     Log::Info << "Saving dictionary matrix to '" << dictionaryFile << "'.\n";
@@ -141,7 +146,7 @@
     LocalCoordinateCoding<> lcc(input, atoms, lambda);
 
     // Run LCC.
-    lcc.Encode(maxIterations);
+    lcc.Encode(maxIterations, objTolerance);
 
     // Save the results.
     Log::Info << "Saving dictionary matrix to '" << dictionaryFile << "'.\n";