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

Cách vẽ mô hình Keras dưới dạng đồ thị và hiển thị hình dạng đầu vào/đầu ra bằng Python

TensorFlow framework ML của Google. Keras API deep learning Python trên TensorFlow. Functional API xây dựng đồ thị lớp DAG.

Sử dụng Google Colab. Code vẽ mô hình với và không với show_shapes:

Ví dụ

print("Vẽ mô hình đã xây dựng")
keras.utils.plot_model(model, "my_model.png")
print("Hiển thị kích thước đầu vào và đầu ra")
keras.utils.plot_model(model, "my_first_model.png", show_shapes=True)

Nguồn: TensorFlow Guide

Đầu ra

Cách vẽ mô hình Keras dưới dạng đồ thị và hiển thị hình dạng đầu vào/đầu ra bằng Python

Giải thích

  • Đồ thị các lớp là hình ảnh tư duy cho mô hình deep learning, tạo được bằng Functional API.
  • Tham số show_shapes=True hiển thị shape của tensor đầu vào/đầu ra từng lớp.