HEXDEV-194: Do regression for GBM/DRF reproducibility tests.

valeyko · Mar 17, 2015 · 70b97a9 · 70b97a9
1 parent d70a4b8
commit 70b97a9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/test/java/hex/drf/DRFTest.java b/src/test/java/hex/drf/DRFTest.java
@@ -252,6 +252,7 @@ public void basicDRF(String fnametrain, String hexnametrain, String fnametest, S
         parms.max_depth = 8;
         parms.mtries = -1;
         parms.min_rows = 10;
+        parms.classification = false;
         parms.seed = 1234;
 
         // Build a first model; all remaining models should be equal

diff --git a/src/test/java/hex/gbm/GBMTest.java b/src/test/java/hex/gbm/GBMTest.java
@@ -383,6 +383,7 @@ public GBMModel basicGBM(String fname, String hexname, PrepData prep, boolean va
         parms.learn_rate = 0.1;
         parms.min_rows = 10;
         parms.family = Family.AUTO;
+        parms.classification = false;
 
         // Build a first model; all remaining models should be equal
         GBMModel gbm = parms.fork().get();