# 定义数据集类 class MyDatasetDataset def __init__self x y selfx = x selfy = y def __getitem__self idx return selfxidx selfyidx def __len__self return lenselfx

这是一个简单的Informer网络，包含一个编码器和解码器，用于处理时间序列数据。编码器和解码器都是由多个线性层和ReLU激活函数组成的神经网络。输入大小为5，输出大小为1，使用均方误差损失函数进行训练。训练数据集和测试数据集都是通过自定义的数据集类MyDataset加载的，并使用PyTorch的数据加载器DataLoader进行批量处理。模型使用Adam优化器进行训练，并在每个epoch结束时保存模型参数。