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文件。

  • 5
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值