模型繪圖工具

[原始碼]

plot_model 函數

keras.utils.plot_model(
    model,
    to_file="model.png",
    show_shapes=False,
    show_dtype=False,
    show_layer_names=False,
    rankdir="TB",
    expand_nested=False,
    dpi=200,
    show_layer_activations=False,
    show_trainable=False,
    **kwargs
)

將 Keras 模型轉換為 dot 格式並儲存到檔案。

範例

inputs = ...
outputs = ...
model = keras.Model(inputs=inputs, outputs=outputs)

dot_img_file = '/tmp/model_1.png'
keras.utils.plot_model(model, to_file=dot_img_file, show_shapes=True)

引數

• model：一個 Keras 模型實例
• to_file：繪圖影像的檔案名稱。
• show_shapes：是否顯示形狀資訊。
• show_dtype：是否顯示層 dtype。
• show_layer_names：是否顯示層名稱。
• rankdir：傳遞給 PyDot 的 rankdir 引數，一個字串，指定繪圖的格式："TB" 建立垂直繪圖；"LR" 建立水平繪圖。
• expand_nested：是否將巢狀 Functional 模型展開為叢集。
• dpi：影像解析度，單位為每英吋點數。
• show_layer_activations：顯示層啟動 (僅適用於具有 activation 屬性的層)。
• show_trainable：是否顯示層是否可訓練。

回傳

如果已安裝 Jupyter，則回傳 Jupyter notebook Image 物件。這可以在 notebook 中啟用模型繪圖的內嵌顯示。

[原始碼]

model_to_dot 函數

keras.utils.model_to_dot(
    model,
    show_shapes=False,
    show_dtype=False,
    show_layer_names=True,
    rankdir="TB",
    expand_nested=False,
    dpi=200,
    subgraph=False,
    show_layer_activations=False,
    show_trainable=False,
    **kwargs
)

將 Keras 模型轉換為 dot 格式。

引數

• model：一個 Keras 模型實例。
• show_shapes：是否顯示形狀資訊。
• show_dtype：是否顯示層 dtype。
• show_layer_names：是否顯示層名稱。
• rankdir：傳遞給 PyDot 的 rankdir 引數，一個字串，指定繪圖的格式："TB" 建立垂直繪圖；"LR" 建立水平繪圖。
• expand_nested：是否將巢狀 Functional 模型展開為叢集。
• dpi：影像解析度，單位為每英吋點數。
• subgraph：是否回傳 pydot.Cluster 實例。
• show_layer_activations：顯示層啟動 (僅適用於具有 activation 屬性的層)。
• show_trainable：是否顯示層是否可訓練。

回傳

一個 pydot.Dot 實例，代表 Keras 模型；或一個 pydot.Cluster 實例，如果 subgraph=True，則代表巢狀模型。