GH-15809: add AIC and Loglikelihood to output metrics

h2oai · Jan 17, 2024 · 8f66d3e · 8f66d3e
1 parent 566eb00
commit 8f66d3e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/h2o-core/src/main/java/water/api/schemas3/ModelMetricsBinomialV3.java b/h2o-core/src/main/java/water/api/schemas3/ModelMetricsBinomialV3.java
@@ -57,6 +57,8 @@ public S fillFromImpl(ModelMetricsBinomial modelMetrics) {
 //    sigma = modelMetrics._sigma;
     r2 = modelMetrics.r2();
     logloss = modelMetrics._logloss;
+    loglikelihood = modelMetrics._loglikelihood;
+    AIC = modelMetrics._aic;
     mean_per_class_error = modelMetrics._mean_per_class_error;
 
 

diff --git a/h2o-py/h2o/model/metrics_base.py b/h2o-py/h2o/model/metrics_base.py
@@ -151,7 +151,10 @@ def _str_items(self, verbosity=None):
                     "Null deviance: {}".format(self.null_deviance()),
                     "Residual deviance: {}".format(self.residual_deviance()),
                 ])
-                if is_type(self.aic(), numeric): items.append("AIC: {}".format(self.aic()))
+        if is_type(self.aic(), numeric) and self.loglikelihood() != 0:
+            items.append("AIC: {}".format(self.aic()))
+        if is_type(self.loglikelihood(), numeric) and self.loglikelihood() != 0:
+            items.append("Loglikelihood: {}".format(self.loglikelihood()))
         items.extend(self._str_items_custom())
         return items