TensorFlow keras plot_mode 打印模型,解决报错的问题

TensorFlow keras plot_mode 打印模型,解决报错的问题

tf.keras.utils.plot_model(model,to_file=‘model1.png’,show_shapes=True,show_layer_names=True,rankdir=‘TB’)

1、安装
pip3 install pydot
pip3 install graphviz
2、修改环境变量
1、ctrl + Alt + T:打开命令行
2、输入:
sudo gedit ~/.bashrc
在文件末尾编辑:
export PATH=/usr/bin/dot:$PATH,保存,退出
3、source ~/.bashrc

你可能感兴趣的:(TensorFlow keras plot_mode 打印模型,解决报错的问题)