我想训练一个RNN,其隐藏状态激活大小与Keras默认提供的不同。例如:我的输入向量在每个时间步长为27,输出也是27。我希望隐藏状态激活大小为50。
回答:
你所说的隐藏状态激活大小具体指什么并不清楚。在Keras中,控制RNN单元大小的参数称为units
。根据文档:
- units: 正整数,输出空间的维度。
这个数字直接对应于单元内部应用的递归矩阵的形状,因此在某种意义上它是单元的隐藏大小,或者是隐藏神经元的数量。
要将这个大小从27
更改为50
,只需调用:
model.add(SimpleRNN(50, ...))