ImportError：无法导入名称“model_to_dot”

Posted 2023-02-23

【中文标题】ImportError：无法导入名称“model_to_dot”

【英文标题】：ImportError: cannot import name 'model_to_dot'

【发布时间】：2019-06-15 20:44:35

【问题描述】：

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

【参考方案1】：

如果您将第 2 行更改为以下内容，则可以导入 model_to_dot：

from keras.utils.vis_utils import model_to_dot

【讨论】：

@SaravanabalagiRamachandran，文档是最新的，它们反映了最新的 API 状态。但是，截至目前，最新的 keras 版本已经相当老了 2.2.4（2018 年 10 月 3 日）。这里，是对应的commit message。所以他们只需要再次发布！

【参考方案2】：

如果您将第 2 行更改为以下内容，则可以导入 model_to_dot：

from tensorflow.keras.utils import model_to_dot