Cách lưu và 직렬 hóa mô hình Keras bằng Python

TensorFlow framework ML của Google. Tensor là mảng đa chiều. Keras API deep learning Python trên TensorFlow, mở rộng cao, chạy TPU/GPU, xuất ra web/mobile.

Sử dụng Google Colab. Code lưu mô hình, xóa, sau đó tải lại:

Ví dụ

print("Lưu mô hình vào file")
model.save("path_to_my_model")
print("Xóa mô hình")
del model
print("Tái tạo mô hình từ file đã lưu")
model = keras.models.load_model("path_to_my_model")

Nguồn: TensorFlow Guide

Đầu ra

Lưu mô hình vào file
INFO:tensorflow:Assets written to: path_to_my_model/assets
Xóa mô hình
Tái tạo mô hình từ file đã lưu

Giải thích

• Lưu mô hình bằng model.save.
• Xóa mô hình để minh họa tải lại.
• Tải lại bằng keras.models.load_model.