怎么用python进行回归预测_pytorch-RNN进行回归曲线预测方式

任务

通过输入的sin曲线与预测出对应的cos曲线

#初始加载包 和定义参数

import torch

from torch import nn

import numpy as np

import matplotlib.pyplot as plt

torch.manual_seed(1) #为了可复现

#超参数设定

TIME_SETP=10

INPUT_SIZE=1

LR=0.02

DOWNLoad_MNIST=True

定义RNN网络结构

from torch.autograd import Variable

class RNN(nn.Module):

def __init__(self):

#在这个函数中，两步走，先init,再逐步定义层结构

super(RNN,self).__init__()

self.rnn=nn.RNN( #定义32隐层的rnn结构

input_size=1,

hidden_size=32, #隐层有32个记忆体

num_layers=1, #隐层层数是1

batch_first=True

)

self.out=nn.Linear(32,1) #32个记忆体对应一个输出

def forward(self,x,h_state):

#前向过程，获取 rnn网络输出r_put(注意这里r_out并不是最后输出，最后要经过全连接层) 和 记忆体情况h_state

r_out,h_state=self.rnn(x,h_state)