如何使用Keras更改RNN中的隐藏状态激活大小?

时间:2018-04-13 20:47:25

标签: machine-learning neural-network deep-learning keras rnn

我想训练一个RNN,其隐藏状态激活的大小与keras中默认提供的大小不同。例如:我的输入向量在时间步长为27,输出也是27.我希望隐藏状态激活大小为50。

1 个答案:

答案 0 :(得分:1)

目前尚不清楚隐藏状态激活大小究竟是什么意思。控制keras中RNN单元大小的参数称为units。来自documentation

  
      
  • 单位:输出空间的正整数,维数。
  •   

此数字直接对应于在单元格内应用的循环矩阵的形状,因此从这个意义上说,它是单元格的隐藏大小,或隐藏神经元的数量

要将此尺寸从27更改为50,只需致电:

model.add(SimpleRNN(50, ...))