remove generations=100 from tpot

zuliani99 · Jul 30, 2021 · 72f8246 · 72f8246
1 parent 9e96acb
commit 72f8246
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app/algorithms/tpot.py b/app/algorithms/tpot.py
@@ -16,7 +16,7 @@
 def make_classification(X_train, X_test, y_train, y_test, timelife, y, client, time_start):
 
   # Model Classification
-  model =  TPOTClassifier(generations=100, cv=10, max_time_mins=timelife, random_state=1, verbosity=2, n_jobs=-1, max_eval_time_mins=5)
+  model =  TPOTClassifier(cv=10, max_time_mins=timelife, random_state=1, verbosity=2, n_jobs=-1, max_eval_time_mins=5)
   #model.fit(X_train, y_train)
 
   with joblib.parallel_backend("dask"):
@@ -41,7 +41,7 @@ def make_classification(X_train, X_test, y_train, y_test, timelife, y, client, t
 def make_regression(X_train, X_test, y_train, y_test, timelife, client, time_start):
 
   # Regression model
-  model =  TPOTRegressor(generations=100, cv=10, max_time_mins=timelife, random_state=1, verbosity=2, n_jobs=-1, max_eval_time_mins=5)
+  model =  TPOTRegressor(cv=10, max_time_mins=timelife, random_state=1, verbosity=2, n_jobs=-1, max_eval_time_mins=5)
   #model.fit(X_train, y_train)
 
   with joblib.parallel_backend("dask"):