pytorch生成带滑动的训练集测试集（batch)

沉思的雅典娜

已于 2022-04-14 17:31:24 修改

阅读量1.6k

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch batch lstm 文章标签： pytorch LSTM RNN batch

于 2022-04-14 17:16:46 首次发布

本文链接：https://blog.youkuaiyun.com/cf_jack/article/details/124175312

本文介绍了如何在PyTorch中创建带有滑动步长的训练集和测试集，特别适用于LSTM等需要序列输入的模型。通过示例代码展示了不同滑动步长下批处理的生成过程，解决训练数据的连续性问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

话不多说，先上代码

Example1

from torch.utils import data
import torch

batch_size = 3
t1 = np.arange(0,20)
t2 = np.arange(0,20)+100

生成序列0~19和100~190，分别模拟x_train和y_train

tensor1 = torch.from_numpy(t1)
tensor2 = torch.from_numpy(t2)

上一步很关键，否则报错

data_set = data.TensorDataset(tensor1,tensor2)
data_iter = data.DataLoader(data_set,batch_size=3,shuffle=False)
for x_batch, y_batch in data_iter:
    print('-----------------')
    dsp(x_batch)
    dsp(y_batch)
    print('-----------------')

部分输出结果：

不过上面的例子有个问题，如果使用类似LSTM这种模型，需要滑动输入比如

原始如下：

tensor([0, 1, 2])

tensor([100, 101, 102])

tensor([3, 4, 5])

最低0.47元/天解锁文章

200万优质内容无限畅学