Przykład oddzwonienia Keras do zapisywania modelu po każdej epoce?

17

Czy ktoś może napisać prosty przykład Keras wykorzystujący wywołanie zwrotne, aby zapisać model po każdej epoce? Potrafię znaleźć przykłady oszczędzania ciężarów, ale chcę móc zapisać w pełni funkcjonujący model po każdej epoce treningowej.

I_Play_With_Data
źródło

Odpowiedzi:

22

Ustawienie „save_weights_only” na False w wywołaniu zwrotnym Keras „ModelCheckpoint” spowoduje zapisanie pełnego modelu; ten przykład wzięty z powyższego linku zapisze pełny model w każdej epoce, niezależnie od wydajności:

keras.callbacks.ModelCheckpoint(filepath, monitor='val_loss', verbose=0, save_best_only=False, save_weights_only=False, mode='auto', period=1)

Więcej przykładów można znaleźć tutaj , w tym oszczędność tylko ulepszonych modeli oraz wczytywania zapisanych modeli.

redhqs
źródło
10

Pamiętaj, aby uwzględnić zmienną epoki w ścieżce pliku. W przeciwnym razie zapisany model zostanie wymieniony po każdej epoce.

filepath = "saved-model-{epoch:02d}-{val_acc:.2f}.hdf5"
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=False, mode='max')

Aby uzyskać więcej przykładów, sprawdź tutaj .

Safwan
źródło
2
Witamy na stronie! I dzięki, doceniam ten dodatek do odpowiedzi
I_Play_With_Data