Update src/mlpack/tests/bayesian_linear_regression_test.cpp
Co-authored-by: Marcus Edel <marcus.edel@fu-berlin.de>
This commit is contained in:
@@ -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);
|
||||
|
||||
Reference in New Issue
Block a user