from keras.utils.visualize_util import plot
ImportError: No module named visualize_util
出現(xiàn)這種錯誤，如何解決呢拒秘？
這是Keras升級到2后出現(xiàn)的問題，因?yàn)榻涌诟淖兞恕?/p>
將代碼改為：
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')
結(jié)果是生成一個model.png
