【Tensorflow+keras】用代码给神经网络结构绘图

小灰灰 2022-10-14 10:55 89阅读 0赞

# 1 引言 #

在论文中可能有必要讲述清楚网络的结构，可以用代码实现漂亮的绘制结构。如图所示。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MzkzNTY5Ng_size_16_color_FFFFFF_t_70_pic_center]

# 2 步骤 #

安装两个包，本人是LInux系统，用以下命令，如果是其他系统，参考[网址][Link 1]

> sudo apt install graphviz  
> pip install pydot

# 3 实现 #

import tensorflow as tf
    from tensorflow.keras.layers import *
    from tensorflow.keras.models import Sequential, Model
    
    model = tf.keras.Sequential()
    model.add(Conv1D(8,5,strides=2,padding='same',input_shape=(4,4,8)))
    model.add(LeakyReLU())
    model.add(Dropout(0.3))
    
    model.add(Conv1D(16,5, strides=2, padding='same'))
    model.add(LeakyReLU())
    model.add(Dropout(0.3))
    model.add(Flatten())
    model.add(Dense(1,activation='softmax'))
    tf.keras.utils.plot_model(model, to_file='model.png', show_shapes=True)

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MzkzNTY5Ng_size_16_color_FFFFFF_t_70_pic_center]: /images/20221014/6e25deb8bfc347b6b25b9a261966caf6.png
[Link 1]: https://graphviz.gitlab.io/download/#linux

发表评论取消回复

表情：

评论列表（有 0 条评论，89人围观）

还没有评论，来说两句吧...

相关阅读

相关用Python实现神经网络（附完整代码）！

在学习神经网络之前，我们需要对神经网络底层先做一个基本的了解。我们将在本节介绍感知机、反向传播算法以及多种梯度下降法以给大家一个全面的认识。【----帮助Python学习...

迈不过友情╰/ 2024年04月24日 23:21/ 0 赞/ 79 阅读

相关用Java写一段神经网络代码

好的，这是用 Java 写的一段简单的神经网络代码的例子。这个神经网络有两个输入层单元、一个隐藏层单元、一个输出层单元，并且使用 sigmoid 激活函数。 \`\`\` i

╰+攻爆jí腚メ/ 2024年03月27日 17:55/ 0 赞/ 79 阅读

相关用文字描述神经网络的各个结构

神经网络是一种模拟人脑的人工智能模型，它通过设计多个节点来实现对数据的预测和分类。下面是神经网络的几个主要结构： 1. 输入层：输入层是神经网络的第一层，它接收外部数据，并

╰+攻爆jí腚メ/ 2024年03月25日 04:29/ 0 赞/ 47 阅读

相关神经网络结构及python代码实现

神经网络结构及python代码实现在机器学习中，神经网络通常用来学习输入与输出之间的映射关系。在这里，我们介绍一个单隐层神经网络，包含50个隐藏单元和一个3个神经元的输出层

分手后的思念是犯贱/ 2024年03月17日 12:24/ 0 赞/ 49 阅读

相关图神经网络的基本结构

文章目录图神经网络的基本结构图谱和图傅里叶变换基于频谱域的GNN和基于空间域的GNN的比较图神经网络的任务需求和模型要求任务

素颜马尾好姑娘i/ 2023年10月12日 16:20/ 0 赞/ 43 阅读

相关卷积神经网络基本结构介绍以及代码实现

1、基本结构 ![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9

一时失言乱红尘/ 2023年01月13日 15:53/ 0 赞/ 90 阅读

相关神经网络代码注意点

总结文章来源：CSDN\_LawsonAbs LawsonAbs的认知与思考，望各位读者审慎阅读。持续更新~ ------------------

Myth丶恋晨/ 2022年11月19日 05:28/ 0 赞/ 186 阅读

相关【Tensorflow+keras】用代码给神经网络结构绘图

1 引言在论文中可能有必要讲述清楚网络的结构，可以用代码实现漂亮的绘制结构。如图所示。 ![在这里插入图片描述][watermark_type_ZmFuZ3poZW

小灰灰/ 2022年10月14日 10:55/ 0 赞/ 90 阅读

相关 Tensorflow卷积神经网络常用结构

1.卷积层 Tensorflow对卷积神经网络有着很好的支持，下面的程序实现了一个卷积层的前向传播。 import tensorflow as tf

拼搏现实的明天。/ 2022年06月05日 04:39/ 0 赞/ 232 阅读

相关卷积神经网络常见结构

卷积神经网络常见结构 -------------------- 卷积神经网络中常见的层类型有：卷积层（CONV）激活函数（ACT 或 RELU）

忘是亡心i/ 2022年05月29日 00:51/ 0 赞/ 216 阅读