GH-15809: additionally fix aic check in test

h2oai · Feb 5, 2024 · 22c383a · 22c383a
1 parent 6c46408
commit 22c383a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/h2o-py/h2o/model/metrics_base.py b/h2o-py/h2o/model/metrics_base.py
@@ -153,9 +153,9 @@ def _str_items(self, verbosity=None):
                     "Residual deviance: {}".format(self.residual_deviance()),
                 ])
 
-        if is_type(self.aic(), numeric) and not math.isnan(self.aic()) and self.aic() != 0:
+        if (m_is_binomial or m_is_regression or m_is_multinomial) and is_type(self.aic(), numeric) and not math.isnan(self.aic()) and self.aic() != 0:
             items.append("AIC: {}".format(self.aic()))
-        if is_type(self.loglikelihood(), numeric) and not math.isnan(self.loglikelihood()) and self.loglikelihood() != 0:
+        if (m_is_binomial or m_is_regression or m_is_multinomial) and is_type(self.loglikelihood(), numeric) and not math.isnan(self.loglikelihood()) and self.loglikelihood() != 0:
             items.append("Loglikelihood: {}".format(self.loglikelihood()))       
 
         items.extend(self._str_items_custom())