Linux系统中tensorflow 2 模型可视化 pydot

2021-04-21  本文已影响0人  热爱生活H_Li

使用代码:

# 模型构建

model = tf.keras.Model(inputs=[image_input, timeseries_input], outputs=[score_output, class_output])

# 绘制模型数据流图

tf.keras.utils.plot_model(model, "multi_io_model.png", show_shapes=True, dpi=500)

报错信息:

ImportError : Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work.
pydot_ng.InvocationException: GraphViz's executables not found

解决方法:

pip install pydot_ng
conda install python-graphviz
上一篇 下一篇

猜你喜欢

热点阅读