Windows下keras 神经网络模型可视化

最新推荐文章于 2024-10-31 13:42:00 发布

原创

最新推荐文章于 2024-10-31 13:42:00 发布 · 958 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#Keras #Windows #可视化

功能：

Keras中提供了一个神经网络可视化的函数plot，并可以将可视化结果保存在本地

from keras.utils.vis_utils import plot_model
plot_model(model, to_file='model.png')

安装：

首先安装pydot模块（直接利用conda 安装），然后安装graphviz软件（当前使用2.38版本），下载地址为：http://www.graphviz.org/，下载 graphviz-2.38.msi（https://graphviz.gitlab.io/_pages/Download/Download_windows.html）

双击.msi文件一直next完成安装，然后将安装目录下的\bin 添加进Path环境变量

测试代码：

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


d