Fix PUBDEV-1381 Fix: Check that response column is present in the val…

…idation frame.
h2oai · Jun 11, 2015 · fe4dd15 · fe4dd15
1 parent 463a955
commit fe4dd15
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/h2o-core/src/main/java/hex/ModelBuilder.java b/h2o-core/src/main/java/hex/ModelBuilder.java
@@ -396,7 +396,7 @@ public void init(boolean expensive) {
       try {
         String[] msgs = Model.adaptTestForTrain(_train._names, _parms._weights_column, _parms._offset_column, null, _train.domains(), _valid, _parms.missingColumnsType(), expensive, true);
         _vresponse = _valid.vec(_parms._response_column);
-        if (_vresponse == null)
+        if (_vresponse == null && _parms._response_column != null)
           error("_validation_frame", "Validation frame must have a response column '" + _parms._response_column + "'.");
         if (expensive) {
           for (String s : msgs) {