1.安装:
conda install pygot
conda install graphviz
2.代码:(在test.py中)
导入包
from keras.utils.vis_utils import plot_model
在模型后面利用plot_model()函数,如:
model = models.convert_model(model) plot_model(model, to_file=‘model.png‘, show_shapes=True, show_layer_names=True)
原文:https://www.cnblogs.com/iamdongyang/p/11777100.html