tensorflow keras plot

tech2022-09-16  101

首先安装需求库

pip install pydot-ng

https://graphviz.gitlab.io/download/下载grapgviz指定版本,添加进path

cmd命令行要运行一次dot -c 不然会报错

然后使用tf.keras.utils.plot_model(     model, to_file='model.png', show_shapes=False, show_layer_names=True,     rankdir='TB', expand_nested=False, dpi=96 )

就可以得到模型图

最新回复(0)