☑ ImportError: No module named visualize_util

from keras.utils.visualize_util import plot

ImportError: No module named visualize_util

出现这种错误，如何解决呢？

这是Keras升级到2后出现的问题，因为接口改变了。

将代码改为：

from keras.utils.vis_utils import plot_model

原来的plot 更名为 plot_model

测试代码：

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

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')

结果是生成一个model.png