我尝试使用Ipython和keras.utis.vis_utils可视化Keras模型。但是我收到以下错误。
导入错误: 无法导入
pydot
。请安装pydot
.例如,>pip install pydot
.
使用的代码:
例如,模型 = ggv.h5
from IPython.display import SVG
from keras.utils.vis_utils import model_to_dot
SVG(model_to_dot(model).create(prog='dot', format='svg'))
追踪
Traceback (most recent call last):
File "C:UsersVW3ZTWSPycharmProjectsData_Collection_and_learningsvenvlibsite-packagesIPythoncoreinteractiveshell.py", line 2869, in run_code
exec(code_obj, self.user_global_ns, self.user_ns)
File "<ipython-input-49-b1aadf48e3ac>", line 5, in <module>
SVG(model_to_dot(model).create(prog='dot', format='svg'))
File "C:UsersVW3ZTWSPycharmProjectsData_Collection_and_learningsvenvlibsite-packageskerasutilsvis_utils.py", line 55, in model_to_dot
_check_pydot()
File "C:UsersVW3ZTWSPycharmProjectsData_Collection_and_learningsvenvlibsite-packageskerasutilsvis_utils.py", line 20, in _check_pydot
'Failed to import `pydot`. '
ImportError: Failed to import `pydot`. Please install `pydot`. For example with `pip install pydot`.
我尝试卸载pydot,graphviz,然后再次安装它。但错误保持不变。
在这种情况下,可视化模型的最佳解决方案是什么
如果您在 jupyter notebook 中遇到该错误,则有可能您使用的 python 解释器与系统默认值不同。
在笔记本中尝试以下操作:
import sys
%$sys.executable -m pip install pydot
此代码段会将当前 python 解释器的路径传递给终端,并将 pip 作为模块运行以安装库。
查看此链接以获取更多信息
- https://docs.python.org/3/installing/index.html
- https://ipython.readthedocs.io/en/stable/interactive/magics.html
如果您使用的是其他解释器,则首先要发现您使用的是哪一个:print(sys.executable)
然后使用该信息在终端中运行:path_to_the_interpeter -m pip install pydot
另一种方法是从代码中执行此操作:
import subprocess
import sys
def install(library):
subprocess.call([sys.executable, "-m", "pip", "install", library])
try:
import pydot
except ImportError:
install('pydot')
import pydot
但我确实更喜欢从终端安装库并保持我的代码干净。