From 469ef5cea5dad3d3ced18eaa3909f87f781f50af Mon Sep 17 00:00:00 2001 From: cmercier Date: Thu, 23 Jul 2020 11:22:16 +0200 Subject: [PATCH] Formatting. --- .../bayesian_linear_regression.cpp | 6 +++--- src/mlpack/tests/bayesian_linear_regression_test.cpp | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/mlpack/methods/bayesian_linear_regression/bayesian_linear_regression.cpp b/src/mlpack/methods/bayesian_linear_regression/bayesian_linear_regression.cpp index 472c1638d9..f9af417729 100644 --- a/src/mlpack/methods/bayesian_linear_regression/bayesian_linear_regression.cpp +++ b/src/mlpack/methods/bayesian_linear_regression/bayesian_linear_regression.cpp @@ -126,7 +126,7 @@ double BayesianLinearRegression::CenterScaleData(const arma::mat& data, { if (!centerData && !scaleData) { - dataProc = arma::mat(const_cast(data.memptr()), data.n_rows, + dataProc = arma::mat(const_cast(data.memptr()), data.n_rows, data.n_cols, false, true); responsesProc = arma::rowvec(const_cast(responses.memptr()), responses.n_elem, false, @@ -167,7 +167,7 @@ void BayesianLinearRegression::CenterScaleDataPred( { if (!centerData && !scaleData) { - dataProc = arma::mat(const_cast(data.memptr()), data.n_rows, + dataProc = arma::mat(const_cast(data.memptr()), data.n_rows, data.n_cols, false, true); } @@ -177,7 +177,7 @@ void BayesianLinearRegression::CenterScaleDataPred( } else if (!centerData && scaleData) - { + { dataProc = data.each_col() / dataScale; } diff --git a/src/mlpack/tests/bayesian_linear_regression_test.cpp b/src/mlpack/tests/bayesian_linear_regression_test.cpp index 19c5507dae..344a3130c1 100644 --- a/src/mlpack/tests/bayesian_linear_regression_test.cpp +++ b/src/mlpack/tests/bayesian_linear_regression_test.cpp @@ -180,14 +180,14 @@ BOOST_AUTO_TEST_CASE(EqualtoRidge) if (arma::norm(blrPred - ridgePred) > 1e-5) continue; - // Check the predictions are close enough between ridge and our tested model. + // Check the predictions are close enough between ridge and our blr. for (size_t i = 0; i < y.size(); ++i) BOOST_REQUIRE_CLOSE(blrPred[i], ridgePred[i], 1); // Exit once a test case has completed. break; } - + BOOST_REQUIRE_LT(trial, 3); }