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

Cách vẽ mô hình Keras bằng Python

Keras (tiếng Hy Lạp: "sừng") là API học sâu viết bằng Python, chạy trên TensorFlow. Hỗ trợ thử nghiệm nhanh, mở rộng cao, đa nền tảng.

Functional API giúp xây dựng đồ thị các lớp (DAG) linh hoạt hơn Sequential API.

Sử dụng Google Colab. Code vẽ mô hình dưới đây:

Ví dụ

print("Đang vẽ mô hình")
keras.utils.plot_model(model, "my_resnet.png", show_shapes=True)

Nguồn: TensorFlow Guide

Đầu ra

Cách vẽ mô hình Keras bằng Python

Giải thích

  • Phương thức plot_model vẽ các lớp của mô hình dưới dạng đồ thị.