问题描述
在 Jupyter 中使用如下代码绘制神经网络模型时,发现水平方向只显示了部分图片,效果图如下。
from IPython.display import SVG # 实现神经网络结构的图形化显示
from tensorflow.keras.utils import model_to_dot
SVG(model_to_dot(cnn).create(prog='dot', format='svg')) # 绘图
问题解决
修改代码为如下所示(添加 show_shapes= True, show_layer_names=True, dpi=65
):
from IPython.display import SVG # 实现神经网络结构的图形化显示
from tensorflow.keras.utils import model_to_dot
SVG(model_to_dot(cnn, show_shapes= True, show_layer_names=True, dpi=65).create(prog='dot', format='svg')) # 绘图
效果图如下: