Update src/mlpack/tests/bayesian_linear_regression_test.cpp

Co-authored-by: Marcus Edel <marcus.edel@fu-berlin.de>
This commit is contained in:
mercierc
2020-06-23 17:42:03 +02:00
committed by GitHub
co-authored by Marcus Edel
parent 47d5d1f6e7
commit 342ccdbaec
@@ -170,10 +170,7 @@ BOOST_AUTO_TEST_CASE(EqualtoRidge)
BayesianLinearRegression blr(false, false);
blr.Train(matX, y);
LinearRegression ridge(matX,
y,
blr.Alpha() / blr.Beta(),
false);
LinearRegression ridge(matX, y, blr.Alpha() / blr.Beta(), false);
blr.Predict(matX, blrPred);
ridge.Predict(matX, ridgePred);