python - 导入错误 : cannot import name 'model_to_dot'

Question

我想使用 Google Colab 环境可视化 Keras 模型。我找到了 this link .但是，如果我将此链接中的代码复制粘贴到 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