Early Stopper

Classes:

EarlyStopper([stopping, patience, …])

Class for early stopping a models training.

class early_stopper.EarlyStopper(stopping=True, patience=1, min_delta=0.0, minimise=True)

Bases: object

Class for early stopping a models training.

Methods:

check_early_stop(metric_val)

Checks to see if training should stop early based on lack of improvement in supplied metric value.

check_early_stop(metric_val)

Checks to see if training should stop early based on lack of improvement in supplied metric value.

Parameters

metric_val (float) – The current metric value to compare

Returns

True if training should stop (no improvement for ‘patience’ number of epochs, else False

Return type

(bool)