Update test_e2e.py

BayesWitnesses · May 29, 2020 · 809c91e · 809c91e
1 parent a1870e0
commit 809c91e
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/tests/e2e/test_e2e.py b/tests/e2e/test_e2e.py
@@ -37,47 +37,47 @@
 
 
 # Set of helper functions to make parametrization less verbose.
-def regression(model, test_fraction=0.5):
+def regression(model, test_fraction=0.8):
     return (
         model,
         utils.get_regression_model_trainer(test_fraction),
         REGRESSION,
     )
 
 
-def classification(model, test_fraction=0.5):
+def classification(model, test_fraction=0.8):
     return (
         model,
         utils.get_classification_model_trainer(test_fraction),
         CLASSIFICATION,
     )
 
 
-def classification_binary(model, test_fraction=0.5):
+def classification_binary(model, test_fraction=0.8):
     return (
         model,
         utils.get_binary_classification_model_trainer(test_fraction),
         CLASSIFICATION,
     )
 
 
-def regression_random(model, test_fraction=0.5):
+def regression_random(model, test_fraction=0.8):
     return (
         model,
         utils.get_regression_random_data_model_trainer(test_fraction),
         REGRESSION,
     )
 
 
-def classification_random(model, test_fraction=0.5):
+def classification_random(model, test_fraction=0.8):
     return (
         model,
         utils.get_classification_random_data_model_trainer(test_fraction),
         CLASSIFICATION,
     )
 
 
-def classification_binary_random(model, test_fraction=0.5):
+def classification_binary_random(model, test_fraction=0.8):
     return (
         model,
         utils.get_classification_binary_random_data_model_trainer(
@@ -86,7 +86,7 @@ def classification_binary_random(model, test_fraction=0.5):
     )
 
 
-def regression_bounded(model, test_fraction=0.5):
+def regression_bounded(model, test_fraction=0.8):
     return (
         model,
         utils.get_bounded_regression_model_trainer(test_fraction),
@@ -187,11 +187,11 @@ def regression_bounded(model, test_fraction=0.5):
 
         # XGBoost (tree method "hist")
         regression(xgboost.XGBRegressor(**XGBOOST_HIST_PARAMS),
-                   test_fraction=0.5),
+                   test_fraction=0.8),
         classification(xgboost.XGBClassifier(**XGBOOST_HIST_PARAMS),
-                       test_fraction=0.5),
+                       test_fraction=0.8),
         classification_binary(xgboost.XGBClassifier(**XGBOOST_HIST_PARAMS),
-                              test_fraction=0.5),
+                              test_fraction=0.8),
 
         # XGBoost (LINEAR)
         regression(xgboost.XGBRegressor(**XGBOOST_PARAMS_LINEAR)),