Keras提供了一個名為"plot_model"的函數,可以用來可視化模型的結構。使用該函數需要安裝pydot和graphviz兩個庫。
首先,安裝pydot和graphviz庫:
pip install pydot
pip install graphviz
然后,在代碼中可視化模型結構:
from keras.models import load_model
from keras.utils.vis_utils import plot_model
# 加載模型
model = load_model('model.h5')
# 將模型結構可視化為圖片
plot_model(model, to_file='model.png', show_shapes=True)
運行上面的代碼后,會生成一個名為"model.png"的圖片文件,顯示模型的結構。可以根據需要調整參數,比如設置show_shapes參數為True,可以顯示每層的輸入輸出形狀。