Her çağdan sonra bir modeli kaydetmek için Keras Geri Arama örneği?


17

Birisi her çağdan sonra bir modeli kaydetmek için bir geri arama kullanarak doğrudan Keras örneği yayınlayabilir mi? Ağırlık tasarrufu için örnekler bulabilirim, ancak her eğitim çağından sonra tamamen çalışan bir modeli kaydedebilmek istiyorum.

Yanıtlar:


22

'ModelCheckpoint' Keras geri çağrısında 'save_weights_only' değerini False olarak ayarlamak tam modeli kaydeder; yukarıdaki bağlantıdan alınan bu örnek, performanstan bağımsız olarak her çağda tam bir model kaydedecektir:

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

Burada yalnızca geliştirilmiş modellerin kaydedilmesi ve kaydedilen modellerin yüklenmesi de dahil olmak üzere daha fazla örnek bulabilirsiniz .


10

Dosyayolunuza epoch değişkeni eklediğinizden emin olun. Aksi takdirde kaydedilen modeliniz her çağdan sonra değiştirilir.

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

Daha fazla örnek için buraya bakın .


2
Siteye Hoşgeldiniz! Ve teşekkürler, cevaba bu
eki
Sitemizi kullandığınızda şunları okuyup anladığınızı kabul etmiş olursunuz: Çerez Politikası ve Gizlilik Politikası.
Licensed under cc by-sa 3.0 with attribution required.