add additional stress test with more trees

rocker-org · Mar 2, 2019 · 85991f9 · 85991f9
1 parent ae098b8
commit 85991f9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ml/gpu/tests.R b/ml/gpu/tests.R
@@ -40,6 +40,9 @@ h2o.xgboost(y = dependent, x = independent, training_frame = australia,
 h2o.xgboost(y = dependent, x = independent, training_frame = australia,
             ntrees = 1000, backend = "gpu")
 
+## h2o stress test
+h2o.xgboost(y = dependent, x = independent, training_frame = australia,
+            ntrees = 100000, backend = "gpu")
 
 ## Tensorflow: