apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Adjust tolerances to prevent random failures.

Browse Source
This commit is contained in:
Ryan Curtin
2017-12-22 15:32:34 -05:00
parent 2d04abc9fc
commit 60ac702b0f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/tests/momentum_sgd_test.cpp
+2 -2
View File
@@ -65,12 +65,12 @@ BOOST_AUTO_TEST_CASE(GeneralizedRosenbrockTest)
// Create the generalized Rosenbrock function.
GeneralizedRosenbrockFunction f(i);
MomentumUpdate momentumUpdate(0.4);
MomentumSGD s(0.001, 1, 0, 1e-15, true, momentumUpdate);
MomentumSGD s(0.0008, 1, 0, 1e-15, true, momentumUpdate);
arma::mat coordinates = f.GetInitialPoint();
double result = s.Optimize(f, coordinates);
BOOST_REQUIRE_SMALL(result, 1e-10);
BOOST_REQUIRE_SMALL(result, 1e-4);
for (size_t j = 0; j < i; ++j)
BOOST_REQUIRE_CLOSE(coordinates[j], (double) 1.0, 1e-3);
}