keras: как сохранить историю учебных

Question

В Keras, мы можем вернуть выход model.fit к истории следующим образом:

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

Теперь, как сохранить историю в файл для дальнейшего использования (например, рисовать сюжеты acc или loss против эпох)?

Answer 1

A history Объекты имеют поле history - это словарь, который содержит различные учебные показатели, охватываемые каждой учебной эпохой. Так, например, history.history['loss'][99] вернет потерю вашей модели в 100-ю эпоху обучения. Чтобы сохранить это, вы можете pickle этот словарь или просто сохранить различные списки из этого словаря в соответствующий файл.

Answer 2

Что я использую следующее:

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

Таким образом, я сохранить историю как словарь в случае, если я хочу, чтобы построить потери или точность позже.