[GH-5720] Add AIC/Loglikelihood metrics to MetricsTestSuite (#5721)

* fix metrics generation (#5718) (cherry picked from commit 3af804b) * [GH-5720] Add AIC/Loglikelihood metrics to MetricsTestSuite
h2oai · Mar 28, 2024 · b3bef8b · b3bef8b
1 parent 3af804b
commit b3bef8b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ml/src/test/scala/ai/h2o/sparkling/ml/metrics/MetricsTestSuite.scala b/ml/src/test/scala/ai/h2o/sparkling/ml/metrics/MetricsTestSuite.scala
@@ -55,6 +55,8 @@ class MetricsTestSuite extends FunSuite with Matchers with SparkTestContext {
       |
       | "r2" : 1.0,
       | "logloss": 2.0,
+      | "loglikelihood": 2.0,
+      | "AIC": 4.0,
       | "AUC": 3.0,
       | "pr_auc": 4.0,
       | "Gini": 5.0,