要可視化Keras模型的結構,可以使用keras.utils.vis_utils
模塊中的plot_model
函數。該函數可以將模型的結構圖保存為圖片或者顯示在Jupyter Notebook中。
下面是一個示例代碼,用于可視化一個簡單的Keras模型的結構:
from keras.models import Sequential
from keras.layers import Dense
from keras.utils.vis_utils import plot_model
# 創建一個簡單的Sequential模型
model = Sequential()
model.add(Dense(32, input_shape=(784,), activation='relu'))
model.add(Dense(10, activation='softmax'))
# 可視化模型結構并保存為圖片
plot_model(model, to_file='model.png', show_shapes=True)
# 在Jupyter Notebook中顯示模型結構
plot_model(model, show_shapes=True)
運行以上代碼,會生成一個名為model.png
的圖片文件,顯示了模型的結構圖;同時在Jupyter Notebook中顯示模型結構圖。您可以根據實際需求調整模型結構,并根據需要調整plot_model
函數中的參數。