如何用Keras将神经网络体系结构可视化?
我尝试了以下内容:
#!/usr/bin/env python
import keras
from keras.models import model_from_yaml
model_file_path = 'model-301.yaml'
weights_file_path = 'model-301.hdf5'
# Load network
with open(model_file_path) as f:
yaml_string = f.read()
model = model_from_yaml(yaml_string)
model.load_weights(weights_file_path)
model.compile(optimizer='adagrad', loss='binary_crossentropy')
# Visualize
from keras.utils.visualize_util import plot
但是,这给出了一个错误:
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/home/moose/.local/lib/python2.7/site-packages/keras/utils/visualize_util.py", line 7, in <module>
if not pydot.find_graphviz():
AttributeError: 'module' object has no attribute 'find_graphviz'
我怎样才能解决这个问题?
注意:可以在Github上找到hdf5和YAML文件。
该问题也在keras项目的问题页面上引用。 您需要安装pydot
<= 1.1.0版本,因为函数find_graphviz
已在版本1.2.0中删除。 或者,您可以安装pydot-ng,而不是keras开发人员推荐的。
如果你还没有安装pydot
python软件包 - 尝试安装它。 如果你有pydot
重新安装应该帮助你的问题。
上一篇: How can a neural network architecture be visualized with Keras?