- class optuna.study.MaxTrialsCallback(n_trials, states=(TrialState.COMPLETE,))[source]
Set a maximum number of trials before ending the study.
optuna.optimizesets the number of trials that will be run, you may want to continue running until you have a certain number of successfullly completed trials or stop the study when you have a certain number of trials that fail. This
MaxTrialsCallbackclass allows you to set a maximum number of trials for a particular
TrialStatebefore stopping the study.
import optuna from optuna.study import MaxTrialsCallback from optuna.trial import TrialState def objective(trial): x = trial.suggest_float("x", -1, 1) return x**2 study = optuna.create_study() study.optimize( objective, callbacks=[MaxTrialsCallback(10, states=(TrialState.COMPLETE,))], )
n_trials (int) – The max number of trials. Must be set to an integer.
states (Optional[Container[TrialState]]) – Tuple of the
TrialStateto be counted towards the max trials limit. Default value is
None, count all states.