阅读量:0
要可视化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函数中的参数。
