optuna.integration.CatalystPruningCallback

class optuna.integration.CatalystPruningCallback(*args, **kwargs)[source]

Catalyst callback to prune unpromising trials.

Parameters
  • trial – A Trial corresponding to the current evaluation of the objective function.

  • metric (str) – Name of a metric, which is passed to catalyst.core.State.valid_metrics dictionary to fetch the value of metric computed on validation set. Pruning decision is made based on this value.

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, metric='loss')[source]

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(trial[, metric])

Initialize self.

on_epoch_end(state)