tensorflow2 中使用pydot绘制网络模型结构图

主题:pydot,环境tensorflow2.0

  1. 安装pydot之前,要事先安装anaconda
  2. conda命令行当中执行命令,它会自动去下载pydot和graphviz等相关依赖包的
	conda install pydot
	或者使用清华镜像源,速度快
	pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pydot
  1. 安装完成后,执行命令查看已安装的包,找一下有没有pydot包,判断是否安装成功
   	conda list
  1. 如果这个时候,你已经打开了pycharm或者jupyter notebook等编译器,最好重启一下,这样能够自动刷新安装的包,才可以正常使用

关于pydot的使用

1. 语法参数

 tf.keras.utils.plot_model(
   	 model,
   	 to_file='model.png',
  	 show_shapes=False,
   	 show_layer_names=True,
     rankdir='TB',
  	 expand_nested=False,
   	 dpi=96
)

2. 使用方式

在创建神经网络模型model后,一般是通过mode.summary()可以查看网络的模型,如果模型太长太复杂的时候,这种方式看起来不是很直观,可以在模型model创建完成后,使用

导包+数据加载代码
模型创建代码   (以简单的线性层为例)
 	model = tf.keras.Sequential([
   		 tf.keras.layers.Dense(10, input_shape=(3,), activation='relu'),
    	 tf.keras.layers.Dense(1)
	])
    tf.keras.utils.plot_model(model, to_file='dense.png', show_shapes=True, show_layer_names=True)
 
在你的程序文件目录下会有一张dense.png,这就是你创建的模型结构

画出的网络结构图还是比较美观的
在这里插入图片描述

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值