Fix survfit logic when there are no numeric predictors.

AI-Cdrone · Oct 23, 2014 · 110db53 · 110db53
1 parent eb1b1f2
commit 110db53
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/hex/CoxPH.java b/src/main/java/hex/CoxPH.java
@@ -196,7 +196,7 @@ protected float[] score0(double[] data, float[] preds) {
       }
       for (int j = n_cats; j < n_data; ++j)
         numsHasNA |= Double.isNaN(data[j]);
-      if (numsHasNA || (catsHasNA && !catsAllNA) || (n_num == 0 && catsAllNA)) {
+      if (numsHasNA || (catsHasNA && !catsAllNA) || (n_num > 0 && catsAllNA)) {
         for (int i = 1; i <= 2 * n_time; ++i)
           preds[i] = Float.NaN;
       } else {