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.
Methods
on_epoch_end
(state)