trw.callbacks.callback_early_stopping

Module Contents

Classes

CallbackEarlyStopping

Use historical runs to evaluate if a run is promising. If not, early stop will raise ExceptionAbortRun

Attributes

logger

trw.callbacks.callback_early_stopping.logger

class trw.callbacks.callback_early_stopping.CallbackEarlyStopping(store: trw.hparams.RunStore, loss_fn: Callable[[trw.basic_typing.HistoryStep], float], raise_stop_fn: Optional[Callable[[float, trw.basic_typing.History], Tuple[bool, str]]] = None, checkpoints: Sequence[float] = (0.1, 0.25, 0.5, 0.75), discard_if_among_worst_X_performers: float = 0.6, only_consider_full_run: bool = True, min_number_of_runs: int = 10)

Bases: trw.callbacks.callback.Callback

Use historical runs to evaluate if a run is promising. If not, early stop will raise ExceptionAbortRun

_initialize(self, num_epochs: int) → None

__call__(self, options, history: trw.basic_typing.History, model, **kwargs)

__del__(self)