[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 (cherry picked from commit b3bef8b)
h2oai · Mar 28, 2024 · 717755b · 717755b
1 parent c57e776
commit 717755b
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,