diff --git a/workflow.py b/workflow.py index 91fb354..fed4cf3 100644 --- a/workflow.py +++ b/workflow.py @@ -43,7 +43,7 @@ def kfpipeline(): name="train", params={"sample" : -1, "label_column" : LABELS, - "test_size" : 0.10}, + "test_size" : 0.20}, hyperparams={'model_pkg_class': ["sklearn.ensemble.RandomForestClassifier", "sklearn.linear_model.LogisticRegression", "sklearn.ensemble.AdaBoostClassifier"]},