diff --git a/h2o-core/src/main/java/hex/ModelMetricsBinomial.java b/h2o-core/src/main/java/hex/ModelMetricsBinomial.java index 206d0fd4e27e..20374d054e22 100755 --- 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> 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 index ab6fba09b45a..69b17cf9b8d1 100755 --- 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 index ce7b2956c136..1ccb0d509322 100755 --- 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); }