1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Úvod do hlubokého učení s Keras
Connected

Cvičení

Předčasné zastavení trénování modelu

Callback pro předčasné zastavení (early stopping) se hodí v situaci, kdy chceš ukončit trénování modelu ve chvíli, kdy se přestane zlepšovat – a to po zadaném počtu epoch. Abys tuto funkci využil/a, musíš callback předat jako seznam do parametru callbacks metody .fit().

Model, který jsi sestavil/a pro detekci padělaných dolarových bankovek, je připravený k trénování – tentokrát s early stoppingem. K dispozici máš také X_train, y_train, X_test a y_test.

Pokyny

100 XP
  • Importuj callback EarlyStopping z tensorflow.keras.callbacks.
  • Definuj callback, který sleduje 'val_accuracy' s hodnotou patience nastavenou na 5 epoch.
  • Natrénuj model s využitím tohoto early stopping callbacku.