Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Cách biên dịch mô hình Sequential đã xây dựng với Keras trong Python

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.