ImportError: No module named 'keras.utils.visualize_util'

最新推荐文章于 2024-04-24 18:45:47 发布

原创最新推荐文章于 2024-04-24 18:45:47 发布 · 4k 阅读

2 ·

CC 4.0 BY-SA版权

问题专栏收录该内容

14 篇文章

订阅专栏

本文详细介绍了当在Keras中遇到'ImportError: No module named 'keras.utils.visualize_util''错误时的解决方案，包括更新模块引用、安装必要的外部包如pydot和graphviz，并提供了一个测试代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

当导入keras.utils.visualize_util遇到：ImportError: No module named 'keras.utils.visualize_util'

原因：keras升级了

from keras.utils.visualize_util import plot

改为

from keras.utils.vis_utils import plot_model

注意：可能需要安装pydot和graphviz包

系统：Ubuntu16.04下安装

pip install pydot, pydot-ng, graphviz

sudo apt-get install graphviz

重启jupyter notebook

(用pip 安装有时可能会出现问题)

通过以下代码进行测试，测试通过，会在目录下生成一个png文件

import numpy as np
from keras.models import Sequential
from keras.layers.core import Dense, Activation
from keras.optimizers import SGD
from keras.utils import np_utils    
from keras.utils.vis_utils import plot_model


def run():
    # 构建神经网络
    model = Sequential()
    model.add(Dense(4, input_dim=2, init='uniform'))
    model.add(Activation('relu'))
    model.add(Dense(2, init='uniform'))
    model.add(Activation('sigmoid'))
    sgd = SGD(lr=0.05, decay=1e-6, momentum=0.9, nesterov=True)
    model.compile(loss='binary_crossentropy', optimizer=sgd, metrics=['accuracy'])

    # 神经网络可视化
    plot_model(model, to_file='model.png')

if __name__ == '__main__':
    run()

参考：https://www.jianshu.com/p/f3a0b2ca0ca8