使用batch_size时LSTM的数据形状问题:1000样本2特征维度确认
LSTM输入形状的判断
首先明确LSTM模型要求的三维输入格式:(batch_size, timesteps, features),三个维度分别代表:
- batch_size:单次迭代处理的样本数量
- timesteps:每个样本包含的时间步长度
- features:每个时间步对应的特征数
回到你的场景:1000个样本(行)、2个特征(列),batch_size设为10。
- 选项
(10,1000,2)意味着每个batch里的10个样本,每个都包含1000个时间步、2个特征。这需要你有10条独立的长时序数据(每条1000步),但你只有1000个样本,显然不符合你的数据情况。 - 合理的处理方式是把1000个样本看作一条连续的长时序,将其切割为长度为10的时间步片段,这样总共能得到100个有效样本。当batch_size=10时,单次迭代会取10个这样的片段,每个片段的形状是
(10,2),所以最终输入形状是**(10,10,2)**。
内容的提问来源于stack exchange,提问作者RAVIKUMAR NADUVIN
相关产品推荐
相关产品推荐

