ImportError：无法导入名称“ model_to_dot”

Question

我想使用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'))

错误：

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'

Answer 1

如果使用以下命令更改第2行，则可以导入model_to_dot：

from keras.utils.vis_utils import model_to_dot