tpot
tpot copied to clipboard
Max timeout
How to set max timeout at which all the trainings stop and save the result?
tpot.TPOTClassifier(generations=100, population_size=100,
offspring_size=None, mutation_rate=0.9,
crossover_rate=0.1,
scoring='accuracy', cv=5,
subsample=1.0, n_jobs=1,
max_time_mins=None, max_eval_time_mins=5,
random_state=None, config_dict=None,
template=None,
warm_start=False,
memory=None,
use_dask=False,
periodic_checkpoint_folder=None,
early_stop=None,
verbosity=0,
disable_update_check=False,
log_file=None
)
Hi @qo4on, according to the TPOT documentation (https://epistasislab.github.io/tpot/api/), it seems that the max_time_mins argument here should implement what you're looking for.
For example, if you want a timeout of 10 minutes, you would do:
tpot.TPOTClassifier(max_time_mins=10)
You can set the other parameters as you wish.