Biên dịch mô hình Sequential với phương thức compile - Keras và Python

Keras là API deep learning Python, chạy trên TensorFlow. Functional API linh hoạt hơn Sequential API.

Sử dụng Google Colab. Code biên dịch mô hình với nhiều hàm mất mát (loss) và trọng số loss khác nhau:

Ví dụ

print("Đang biên dịch mô hình")
model.compile(
    optimizer=keras.optimizers.RMSprop(1e-3),
    loss={
        "priority": keras.losses.BinaryCrossentropy(from_logits=True),
        "department": keras.losses.CategoricalCrossentropy(from_logits=True),
    },
    loss_weights=[1.0, 0.2],
)

Nguồn: TensorFlow Guide

Đầu ra

Đang biên dịch mô hình

Giải thích

• Biên dịch mô hình bằng phương thức compile.
• Có thể gán hàm mất mát khác nhau cho từng đầu ra khi biên dịch.
• Có thể gán trọng số khác nhau cho các loss, điều chỉnh tổng loss huấn luyện.