MODULE

IVORY.CALLBACKS.PRUNING

Pruning class to prune unpromising trials.

Classes
  • Pruning Callback to prune unpromising trials.</>
dataclass

ivory.callbacks.pruning.Pruning(trial=None, metric='')

Callback to prune unpromising trials.

Parameters
  • trial (Trial, optional) A Trial corresponding to the current evaluation of the objective function.
  • metric (str, optional) An evaluation metric for pruning, e.g., val_loss
Attributes
  • metric (str) An evaluation metric for pruning, e.g., val_loss
  • trial (Trial, optional) A Trial corresponding to the current evaluation of the objective function.