[mlpack-git] master: Actually use SGD for the test. (a85e4c2)

gitdub at mlpack.org gitdub at mlpack.org
Mon Feb 22 12:08:49 EST 2016


Repository : https://github.com/mlpack/mlpack
On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/efe49b78c20eea8df5f2d6b47c2024931cb88d8e...0341d4d82c030dd7bcf91f5dfe7b9e452b7b3cdc

>---------------------------------------------------------------

commit a85e4c25591cb1b4c1188d83e99cf659d8ced904
Author: Ryan Curtin <ryan at ratml.org>
Date:   Mon Feb 22 08:22:22 2016 -0800

    Actually use SGD for the test.


>---------------------------------------------------------------

a85e4c25591cb1b4c1188d83e99cf659d8ced904
 src/mlpack/tests/logistic_regression_test.cpp | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/mlpack/tests/logistic_regression_test.cpp b/src/mlpack/tests/logistic_regression_test.cpp
index fd7a21b..f6e853a 100644
--- a/src/mlpack/tests/logistic_regression_test.cpp
+++ b/src/mlpack/tests/logistic_regression_test.cpp
@@ -589,7 +589,8 @@ BOOST_AUTO_TEST_CASE(LogisticRegressionLBFGSGaussianTest)
   }
 
   // Now train a logistic regression object on it.
-  LogisticRegression<> lr(data, responses, 0.5);
+  LogisticRegression<> lr(data.n_rows, 0.5);
+  lr.Train<L_BFGS>(data, responses);
 
   // Ensure that the error is close to zero.
   const double acc = lr.ComputeAccuracy(data, responses);
@@ -635,7 +636,8 @@ BOOST_AUTO_TEST_CASE(LogisticRegressionSGDGaussianTest)
   }
 
   // Now train a logistic regression object on it.
-  LogisticRegression<> lr(data, responses, 0.5);
+  LogisticRegression<> lr(data.n_rows, 0.5);
+  lr.Train<SGD>(data, responses);
 
   // Ensure that the error is close to zero.
   const double acc = lr.ComputeAccuracy(data, responses);




More information about the mlpack-git mailing list