Keras中如何可視化模型結(jié)構(gòu)

小樊
96
2024-03-11 11:54:25

要可視化Keras模型的結(jié)構(gòu),可以使用keras.utils.vis_utils模塊中的plot_model函數(shù)。該函數(shù)可以將模型的結(jié)構(gòu)圖保存為圖片或者顯示在Jupyter Notebook中。

下面是一個(gè)示例代碼,用于可視化一個(gè)簡(jiǎn)單的Keras模型的結(jié)構(gòu):

from keras.models import Sequential
from keras.layers import Dense
from keras.utils.vis_utils import plot_model

# 創(chuàng)建一個(gè)簡(jiǎn)單的Sequential模型
model = Sequential()
model.add(Dense(32, input_shape=(784,), activation='relu'))
model.add(Dense(10, activation='softmax'))

# 可視化模型結(jié)構(gòu)并保存為圖片
plot_model(model, to_file='model.png', show_shapes=True)

# 在Jupyter Notebook中顯示模型結(jié)構(gòu)
plot_model(model, show_shapes=True)

運(yùn)行以上代碼,會(huì)生成一個(gè)名為model.png的圖片文件,顯示了模型的結(jié)構(gòu)圖;同時(shí)在Jupyter Notebook中顯示模型結(jié)構(gòu)圖。您可以根據(jù)實(shí)際需求調(diào)整模型結(jié)構(gòu),并根據(jù)需要調(diào)整plot_model函數(shù)中的參數(shù)。

0