Keras (tiếng Hy Lạp: "sừng") 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 Sequential với list loss và loss_weights:
Ví dụ
print("Đang biên dịch mô hình")
model.compile(
optimizer=keras.optimizers.RMSprop(1e-3),
loss=[
keras.losses.BinaryCrossentropy(from_logits=True),
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
- Mô hình đã xây dựng được biên dịch bằng phương thức
compile. - Có thể gán loss khác nhau cho từng đầu ra.
- Có thể gán trọng số loss khác nhau để điều chỉnh tổng loss huấn luyện.