From 6f98ab7bc9749b04565c512f7eb59c406beb7826 Mon Sep 17 00:00:00 2001 From: Eshaan Agarwal Date: Wed, 13 Apr 2022 20:13:06 +0530 Subject: [PATCH] Fix style issues Co-authored-by: Ryan Curtin --- src/mlpack/methods/linear_regression/linear_regression.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/mlpack/methods/linear_regression/linear_regression.cpp b/src/mlpack/methods/linear_regression/linear_regression.cpp index d046c129a3..774425ebc7 100644 --- a/src/mlpack/methods/linear_regression/linear_regression.cpp +++ b/src/mlpack/methods/linear_regression/linear_regression.cpp @@ -100,7 +100,9 @@ void LinearRegression::Predict(const arma::mat& points, { // We want to be sure we have the correct number of dimensions in the // dataset. - const size_t labels = (parameters.n_rows == 0) ? size_t(0) : size_t(parameters.n_rows - 1); + // Prevent underflow. + const size_t labels = (parameters.n_rows == 0) ? size_t(0) : + size_t(parameters.n_rows - 1); util::CheckSameDimensionality(points, labels, "LinearRegression::Predict()", "points"); // Get the predictions, but this ignores the intercept value