我想使用Google Colab环境可视化Keras模型。我找到了此链接。但是,如果我将代码从此链接复制粘贴到Colab笔记本,则会出现错误:
from IPython.display import SVG
from keras.utils import model_to_dot
SVG(model_to_dot(model).create(prog='dot', format='svg'))
Run Code Online (Sandbox Code Playgroud)
错误:
ImportError Traceback (most recent call last)
<ipython-input-9-819c3bda4aa6> in <module>()
1 from IPython.display import SVG
----> 2 from keras.utils import model_to_dot
3
4 SVG(model_to_dot(model).create(prog='dot', format='svg'))
ImportError: cannot import name 'model_to_dot'
Run Code Online (Sandbox Code Playgroud)