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

使用代码:

# 模型构建

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

你可能感兴趣的:(Linux系统中tensorflow 2 模型可视化 pydot和graphviz)