drugforge.ml.early_stopping

Class for handling early stopping in training.

Classes

BestEarlyStopping(patience[, burnin])

Class for handling early stopping in training based on improvement over best loss.

ConvergedEarlyStopping(n_check, divergence)

Class for handling early stopping in training based on whether loss is still changing.

GeneralizationLossEarlyStopping(alpha[, burnin])

Class for stopping based on the relative increase of val loss at epoch t from lowest val loss up to epoch t.

PatientConvergedEarlyStopping(n_check, ...)

Class for handling early stopping in training based on whether loss is still changing, with patience.

ProgressQuotientEarlyStopping(alpha, k[, burnin])

Class for stopping based on quotient of generalization loss and training progress.

ThresholdEarlyStopping(threshold, patience)

Class for handling early stopping in training based on whether loss has been below a certain threshold for some number of epochs.