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

Formatting.

Browse Source
This commit is contained in:
cmercier
2020-07-23 11:22:16 +02:00
parent 3ee8375e48
commit 469ef5cea5
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/bayesian_linear_regression/bayesian_linear_regression.cpp
+3 -3
View File
@@ -126,7 +126,7 @@ double BayesianLinearRegression::CenterScaleData(const arma::mat& data,
{
if (!centerData && !scaleData)
{
dataProc = arma::mat(const_cast<double*>(data.memptr()), data.n_rows,
dataProc = arma::mat(const_cast<double*>(data.memptr()), data.n_rows,
data.n_cols, false, true);
responsesProc = arma::rowvec(const_cast<double*>(responses.memptr()),
responses.n_elem, false,
@@ -167,7 +167,7 @@ void BayesianLinearRegression::CenterScaleDataPred(
{
if (!centerData && !scaleData)
{
dataProc = arma::mat(const_cast<double*>(data.memptr()), data.n_rows,
dataProc = arma::mat(const_cast<double*>(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;
}
src/mlpack/tests/bayesian_linear_regression_test.cpp
+2 -2
View File
@@ -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);
}