TensorFlow的画图功能非常强大,可以在模型设计完后进行检验、总结与展示。tf调用该函数如下所示:
tf.keras.utils.plot_model(model, to_file='model.png')
初次调用可能会缺少包:
ImportError: Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work.
在Windows环境下,用conda命令安装即可。
conda install pydot
安装效果图:
安装后就可正常使用。此处为在(base)环境下,将Jupyter notebook中kernel设置为Python [conda env:root]:
画出模型结构:
example1:
example2: