1. Học hỏi
  2. /
  3. Khoa Học
  4. /
  5. Nhập môn Deep Learning với Keras

Connected

Bài tập

Kết hợp nhiều callback

Các mô hình deep learning có thể mất nhiều thời gian để huấn luyện, đặc biệt khi bạn dùng kiến trúc sâu hơn và bộ dữ liệu lớn hơn. Việc lưu mô hình mỗi khi nó được cải thiện và dừng lại khi không còn cải thiện giúp bạn bớt lo lắng về việc chọn số epoch để huấn luyện. Bạn cũng có thể khôi phục mô hình đã lưu bất cứ lúc nào và tiếp tục huấn luyện từ nơi đã dừng.

Dữ liệu huấn luyện và kiểm định của mô hình có sẵn trong workspace của bạn dưới dạng X_train, X_test, y_train và y_test.

Hãy sử dụng các callback EarlyStopping() và ModelCheckpoint() để bạn có thể yên tâm đi ăn một hũ bánh quy trong lúc để máy tính làm việc!

Hướng dẫn

100 XP
  • Import cả hai callback EarlyStopping và ModelCheckpoint từ tensorflow.keras.
  • Tạo monitor_val_acc là một callback EarlyStopping để theo dõi 'val_accuracy', với patience là 3 epoch.
  • Tạo model_checkpoint là một callback ModelCheckpoint, lưu mô hình tốt nhất với tên best_banknote_model.hdf5.
  • Fit mô hình của bạn bằng cách cung cấp một danh sách chứa các callback đã định nghĩa và dùng X_test và y_test làm dữ liệu kiểm định.