paulgavrikov/visualkeras

Visualkeras 是一个 Python 软件包，有助于可视化 Keras（独立或包含在 TensorFlow 中）神经网络架构。它允许轻松自定义样式以符合大多数需求。此模块支持层次风格架构生成，非常适合 CNN（卷积神经网络），以及适用于大多数模型（包括简单的前馈网络）的图形风格架构。