bigru参数计算_手工计算深度学习模型中的参数数量

本文介绍了如何手动计算深度学习模型的参数数量,包括前馈神经网络(FFNN),循环神经网络(RNN,如GRU和LSTM)以及卷积神经网络(CNN)。通过示例详细展示了每个模型的参数计算过程。
摘要由CSDN通过智能技术生成
684b6e36a96380fe54bc6e501e881012.png

计算深度学习模型的可训练参数的数量被认为太微不足道了,因为您的代码已经可以为您完成此操作。但是我想在这里留下笔记,让我们偶尔参考一下。以下是我们将要运行的神经网络模型:

  1. 前馈神经网络(FFNN)
  2. 循环神经网络(RNN)
  3. 卷积神经网络(CNN)

我们将使用Keras的API构建机器学习模型,以便于原型设计和干净的代码,因此我们可以在此处快速导入依赖库:

00e01815df8f3c1b0c5644ed14a7c15e.png

构建之后model,调用model.count_params()以验证可训练的参数数量。

1. 前馈神经网络(FFNN)

  • i, 输入大小
  • h,隐藏层的大小
  • o,输出大小

对于一个隐藏层,

num_params

=各层之间的连接+每层中的偏差

= (i×h + h×o)+(h + o)

例1.1:输入大小为3,隐藏层大小为5,输出大小为2

76c8c1c765cd70d004b4a3f34b8f6e02.png

图1.1

  • i = 3
  • h = 5
  • o = 2

num_params

=各层之间的连接+每层中的偏差

= (3×5 + 5×2)+(5 &

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值