Update src/mlpack/tests/softmax_regression_test.cpp

Co-authored-by: Ryan Curtin <ryan@ratml.org>
2024-01-27 16:28:48 +01:00
parent ed6d5dd10c
commit eff0a8818b
1 changed files with 1 additions and 2 deletions
@@ -586,8 +586,7 @@ TEST_CASE("SoftmaxRegressionComputeProbabilitiesTest",
  for (size_t i = 0; i < data.n_cols; ++i)
  {
    REQUIRE(predictions[i] < numClasses);
-    REQUIRE(sum(probabilities.col(i)) ==
-        Approx(1.0).epsilon(1e-7));
+    REQUIRE(sum(probabilities.col(i)) == Approx(1.0).epsilon(1e-7));
  }

  // Test Classify() on a single point.