tpot icon indicating copy to clipboard operation
tpot copied to clipboard

Max timeout

Open qo4on opened this issue 4 years ago • 1 comments

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
                          )

qo4on avatar Apr 21 '21 14:04 qo4on

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.

rachitk avatar Jul 01 '21 15:07 rachitk