GH-15809: add null check

h2oai · Feb 5, 2024 · 8665036 · 8665036
1 parent 1dac03e
commit 8665036
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/h2o-core/src/main/java/hex/ModelMetricsBinomial.java b/h2o-core/src/main/java/hex/ModelMetricsBinomial.java
@@ -207,7 +207,7 @@ public static class MetricBuilderBinomial<T extends MetricBuilderBinomial<T>> ex
         _logloss += w * MathUtils.logloss(err);
       }
 
-      if(m.getClass().toString().contains("Generic")) {
+      if(m != null && m.getClass().toString().contains("Generic")) {
         _loglikelihood += m.likelihood(w, yact[0], ds);
       }
 
@@ -280,7 +280,7 @@ private ModelMetrics makeModelMetrics(Model m, Frame f, GainsLift gl) {
         sigma = weightedSigma();
         mse = _sumsqe / _wcount;
         logloss = _logloss / _wcount;
-        if(m.getClass().toString().contains("Generic")) {
+        if(m != null && m.getClass().toString().contains("Generic")) {
           loglikelihood = -1 * _loglikelihood ; // get likelihood from negative loglikelihood
           aic = m.aic(loglikelihood);
         }

diff --git a/h2o-core/src/main/java/hex/ModelMetricsMultinomial.java b/h2o-core/src/main/java/hex/ModelMetricsMultinomial.java
@@ -315,7 +315,7 @@ public MetricBuilderMultinomial( int nclasses, String[] domain, MultinomialAucTy
       }
 
 
-      if(m.getClass().toString().contains("Generic")) {
+      if(m != null && m.getClass().toString().contains("Generic")) {
         _loglikelihood += m.likelihood(w, yact[0], ds);
       }
       return ds;                // Flow coding
@@ -379,7 +379,7 @@ private void calculateAucsPerRow(double ds[], int iact, double w){
         }
         mse = _sumsqe / _wcount;
         logloss = _logloss / _wcount;
-        if(m.getClass().toString().contains("Generic")) {
+        if(m != null && m.getClass().toString().contains("Generic")) {
           loglikelihood = -1 * _loglikelihood ; // get likelihood from negative loglikelihood
           aic = m.aic(loglikelihood);
         }

diff --git a/h2o-core/src/main/java/hex/ModelMetricsRegression.java b/h2o-core/src/main/java/hex/ModelMetricsRegression.java
@@ -158,7 +158,7 @@ public MetricBuilderRegression(Distribution dist) {
         }
       }
 
-      if(m.getClass().toString().contains("Generic")) {
+      if(m != null && m.getClass().toString().contains("Generic")) {
         _loglikelihood += m.likelihood(w, yact[0], ds);
       }
 
@@ -212,7 +212,7 @@ ModelMetricsRegression computeModelMetrics(Model m, Frame f, Frame adaptedFrame,
       } else {
         meanResDeviance = Double.NaN;
       }
-      if(m.getClass().toString().contains("Generic")) {
+      if(m != null && m.getClass().toString().contains("Generic")) {
         loglikelihood = -1 * _loglikelihood ; // get likelihood from negative loglikelihood
         aic = m.aic(loglikelihood);
       }