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

来源:互联网 发布:ubuntu 14.04 32下载 编辑:程序博客网 时间:2024/05/17 18:00

python2.7和python3.5都可以使用以下代码代替:

 

def run():    # 构建神经网络    model = Sequential()    model.add(Dense(4, input_dim=2, kernel_initializer='uniform'))    model.add(Activation('relu'))    model.add(Dense(2, kernel_initializer='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()



原创粉丝点击