keras画图(可视化)中遇到的问题以及解决办法
程序员文章站
2022-04-26 22:25:40
...
鸣谢http://blog.csdn.net/lk7688535/article/details/52882408
鸣谢http://www.cnblogs.com/combfish/p/6559741.html
官方文档中说直接安装pydot,但是由于keras中使用了pydot.find_graphviz()函数,而这一函数在1.2.*后的版本被剥离了,所以不能直接pip安装pydot,而是安装pydot_ng.'''
Created on 2017年7月15日
备注可以成功的画出图来
@author: huanglei
'''
#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)
上一篇: graphviz 安装和入门