gmolprop.utils.early_stopping.EarlyStopping

class gmolprop.utils.early_stopping.EarlyStopping(patience=10, verbose=False, delta=0, save_path='')

Bases: object

Early stops the training if validation loss doesn’t improve after a given patience.

Methods

__init__	Args:
save_checkpoint	Saves model when validation loss decrease.

__init__(patience=10, verbose=False, delta=0, save_path='') → None

Args:

patience (int): How long to wait after last time validation loss improved.

Default: 7

verbose (bool): If True, prints a message for each validation loss improvement.

Default: False

delta (float): Minimum change in the monitored quantity to qualify as an improvement.

Default: 0

save_checkpoint(val_loss, model) → None

Saves model when validation loss decrease.