Keras: jak zapisać historię szkoleniowych

Question

W Keras, możemy powrócić wyjście model.fit do historii, co następuje:

history = model.fit(X_train, y_train, 
        batch_size=batch_size, 
        nb_epoch=nb_epoch, 
        validation_data=(X_test, y_test)) 

Teraz, jak zapisać historię do pliku w celu dalszych zastosowań (np rysować fabuły acc-loss lub epoki)?

Answer 1

Obiekt history ma pole history, które zawiera różne wskaźniki treningowe rozpięte w każdej epoce treningowej. Np. history.history['loss'][99] zwróci utratę Twojego modelu w setnej epoce szkolenia. Aby zapisać ten słownik, możesz pickle lub zapisać różne listy z tego słownika do odpowiedniego pliku.

Answer 2

Co Używam jest następujące:

with open('/trainHistoryDict', 'wb') as file_pi: 
     pickle.dump(history.history, file_pi) 

W ten sposób mogę zapisać historię, jak w przypadku słownika chcę wykreślić utraty dokładności lub później.