Keras中LSTM参数的含义

units:是输出的维度。  在下图中,中间的绿色cell 里面有四个黄色小框,每一个小黄框代表一个前馈网络层,对,就是经典的神经网络的结构,num_units就是这个层的隐藏神经元个数,就这么简单。其中1、2、4的激活函数是 sigmoid,第三个的激活函数是 tanh。

参考:https://www.zhihu.com/question/64470274/answer/256379387

units还可以理解为:假如units=128,就一个单词而言,你可以把LSTM内部简化看成 ,X为词向量比如64维,W中的128就是units,也就是说通过LSTM,把词的维度由64转变成了128。

参考:https://blog.csdn.net/jiangpeng59/article/details/77646186

input_dim:输入维度,如果第一层是LSTM的话,需要显式设定该值。

return_sequences: 如果为True的话,只返回最后一个状态的输出,是一个(samples,output_dim)2D张量,如果是False,则是返回所有序列状态的输出,是一个(samples,timesteps,output_dim)3D张量。

input_shape:形如(samples,timesteps,input_dim)的3D张量,其中samples是样本个数,timesteps是每个样本的状态序列个数,input_dim是每个状态下的特征数,在实体标注任务当中,假如有10000个句子,每个句子由10个词组成,每个词被映射为100维的词向量。那么输出的格式就为(10000,10,100)。也就是说每个样本中的10个词就是一组序列,每个词都和前后的词有关系。

Keras中LSTM参数的含义_第1张图片

你可能感兴趣的:(数据挖掘&人工智能)