rebase master

h2oai · Aug 23, 2023 · bab109e · bab109e
1 parent b73cc95
commit bab109e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/h2o-genmodel/src/main/java/hex/genmodel/GenModel.java b/h2o-genmodel/src/main/java/hex/genmodel/GenModel.java
@@ -150,7 +150,7 @@ public String[] getOutputNames() {
         outputNames = new String[]{"lp"};
         break;
       case BinomialUplift:
-        outputNames = new String[]{"uplift_predict", "p_y1_ct1", "p_y1_ct0"};
+        outputNames = new String[]{"uplift_predict", "p_y1_with_treatment", "p_y1_without_treatment"};
         break;
       default:
         throw new UnsupportedOperationException("Getting output column names for model category '" +