keras: làm thế nào để lưu lại lịch sử đào tạo

Question

Trong Keras, chúng tôi có thể trả lại sản lượng model.fit đến một lịch sử như sau:

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

Bây giờ, làm thế nào để lưu lại lịch sử vào một tập tin để sử dụng hơn nữa (ví dụ như vẽ âm mưu của acc hoặc mất mát với epochs)?

Answer 1

Một đối tượng history có trường history là từ điển chứa các chỉ số đào tạo khác nhau được trải rộng trên mỗi kỷ nguyên đào tạo. Vì vậy, ví dụ: history.history['loss'][99] sẽ trả về sự mất mát của mô hình của bạn trong giai đoạn đào tạo thứ 100. Để lưu rằng bạn có thể pickle từ điển này hoặc đơn giản lưu các danh sách khác nhau từ từ điển này vào tệp thích hợp.

Answer 2

gì tôi sử dụng như sau:

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

Bằng cách này tôi lưu lịch sử như một cuốn từ điển trong trường hợp tôi muốn âm mưu tổn thất hoặc chính xác sau này.