Callback d'arrêt précoce

🌐

Early stopping callback

Stoppe l'entraînement lorsque le métrique suivie cesse de s'améliorer

Utilisation

luz_callback_early_stopping(
  monitor = "valid_loss",
  min_delta = 0,
  patience = 0,
  mode = "min",
  baseline = NULL
)

Arguments

monitor

Une chaîne au format <set>_<metric> où <set> peut être 'train' ou 'valid' et <metric> est l'abréviation d'une métrique qui est suivie pendant l'entraînement. Le nom de la métrique est insensible à la casse.

min_delta

Amélioration minimale pour réinitialiser le compteur de patience.

patience

Nombre d'époques sans amélioration avant de stopper l'entraînement.

mode

Spécifie la direction considérée comme une amélioration. Par défaut, 'min' est utilisé. Cela peut également être 'max' (lorque on cherche à maximiser la métrique) et 'zero' (plus proche de zéro est mieux).

baseline

Une valeur initiale qui sera utilisée comme la meilleure valeur vue. L'entraînement s'arrêtera si aucune valeur meilleure que la valeur de référence n'est trouvée dans les premières patience époques.

Valeur de retour

Un luz_callback qui déclenche l'arrêt précoce de l'entraînement.

Note

Ce callback ajoute un callback on_early_stopping qui peut être utilisé pour appeler d'autres callbacks dès que l'entraînement s'arrête .

Si verbose=TRUE dans fit.luz_module_generator(), alors un message est imprimé lors de l'arrêt précoce de l'entraînement.

Voir également

Autres luz_callbacks: luz_callback_auto_resume(), luz_callback_csv_logger(), luz_callback_interrupt(), luz_callback_keep_best_model(), luz_callback_lr_scheduler(), luz_callback_metrics(), luz_callback_mixed_precision(), luz_callback_mixup(), luz_callback_model_checkpoint(), luz_callback_profile(), luz_callback_progress(), luz_callback_resume_from_checkpoint(), luz_callback_train_valid(), luz_callback()

Exemples

cb <- luz_callback_early_stopping()
#> Error in luz_callback_early_stopping(): impossible de trouver la fonction "luz_callback_early_stopping"