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);