L'arrêt anticipé : s'arrêter quand la validation ne progresse plus

Partie 6 · ★★★★

Le cours garde les poids de la meilleure époque ; on peut aussi arrêter l'entraînement dès que la validation n'a pas progressé depuis patience époques (variante ajoutée : le notebook entraîne les 60 époques puis recharge les meilleurs poids).

Complète arret_anticipe(val, patience), qui renvoie (meilleure_epoque, epoque_arret) (époques comptées dès 1). Si la patience n'est jamais épuisée, epoque_arret est la dernière époque. Exemple : validation [1.80, 1.20, 1.05, 1.15, 1.40, 1.75].

def arret_anticipe(val, patience):
    meilleure_perte = float("inf")
    meilleure_epoque = 0
    sans_progres = 0
    for epoque, perte_val in enumerate(val, start=1):
        if perte_val < meilleure_perte:
            meilleure_perte = perte_val
            meilleure_epoque = epoque
            sans_progres = ___
        else:
            sans_progres += ___
        if sans_progres >= ___:
            return meilleure_epoque, epoque
    return meilleure_epoque, len(val)

val = [1.80, 1.20, 1.05, 1.15, 1.40, 1.75]

Ctrl/⌘+Entrée