欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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. 

keras画图(可视化)中遇到的问题以及解决办法

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

keras画图(可视化)中遇到的问题以及解决办法

相关标签: graphviz