Keras问题:ImportError: No module named visualize_util

Keras模型可视化时遇到了一个问题(参考可视化),在Anaconda2中启动jupyter notebook并已安装graphviz模块和pydot模块,运行了以下代码:

#encoding=utf-8

from __future__ import absolute_import 
from __future__ import print_function

from keras.utils.visualize_util import plot
from keras.models import Sequential
from keras.layers.core import Dense, Dropout, Activation
from keras.layers.embeddings import Embedding
from keras.layers.recurrent import LSTM

model = Sequential()
model.add(Embedding(input_dim=1024, output_dim=256, input_length=50))
model.add(LSTM(128)) # try using a GRU instead, for fun
model.add(Dropout(0.5))
model.add(Dense(1))
model.add(Activation('sigmoid'))

plot(model, to_file='model1.png',show_shapes=True)
 

出现错误 ImportError: No module named visualize_util
在Keras文件夹的找到utils文件夹,打开发现并没有visualize_util文件,只有一个vis_utils文件,原因就在于Keras的版本,升级到Keras2版本之后,接口发生了变化。

将代码中的相应位置做出修改:

from keras.utils.visualize_util import plot

改为

from keras.utils.vis_utils import plot_model

修改之后再次测试代码,无错误,并且生成了model1.png文件。

发布了14 篇原创文章 · 获赞 11 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/hesongzefairy/article/details/83013004
今日推荐