From 4eec0b8f4617adb6889c8fbb6d43239859dd6432 Mon Sep 17 00:00:00 2001 From: Marcus Edel Date: Thu, 15 Nov 2018 22:32:18 +0100 Subject: [PATCH] Use ConstantStep from ensmallen. --- src/mlpack/tests/bias_svd_test.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/mlpack/tests/bias_svd_test.cpp b/src/mlpack/tests/bias_svd_test.cpp index d4fe3b7841..217ef3b7f4 100644 --- a/src/mlpack/tests/bias_svd_test.cpp +++ b/src/mlpack/tests/bias_svd_test.cpp @@ -343,11 +343,11 @@ BOOST_AUTO_TEST_CASE(BiasSVDFunctionParallelOptimize) // Make the Bias SVD function and the optimizer. BiasSVDFunction biasSVDFunc(data, rank, lambda); - ConstantStep decayPolicy(alpha); + ens::ConstantStep decayPolicy(alpha); // Iterate till convergence. // The threadShareSize is chosen such that each function gets optimized. - ens::ParallelSGD optimizer(0, + ens::ParallelSGD optimizer(0, std::ceil((float) biasSVDFunc.NumFunctions() / omp_get_max_threads()), 1e-5, true, decayPolicy);