code_13.py 198 字节
Newer Older
ToTensor's avatar
ToTensor 已提交
1 2 3
from IPython.display import SVG # 实现神经网络结构的图形化显示
from keras.utils.vis_utils import model_to_dot
SVG(model_to_dot(ann, show_shapes=True).create(prog='dot', format='svg'))