optuna.integration.CatalystPruningCallback¶
- class optuna.integration.CatalystPruningCallback(trial: optuna.trial._trial.Trial, metric: str = 'loss')[source]¶
Catalyst callback to prune unpromising trials.
See the example if you want to add a pruning callback which observes the accuracy of Catalyst’s
SupervisedRunner
.- Parameters
Note
Added in v2.0.0 as an experimental feature. The interface may change in newer versions without prior notice. See https://github.com/optuna/optuna/releases/tag/v2.0.0.
- __init__(trial: optuna.trial._trial.Trial, metric: str = 'loss') None [source]¶
Methods
__init__
(trial[, metric])on_epoch_end
(state)