PyTorch 循环神经网络

循环神经网络是一种以深度学习为导向的算法，它采用顺序方式进行。在神经网络中，我们总是假设每个输入和输出都独立于所有其他层。这种类型的神经网络被称为循环神经网络，因为它们按照顺序进行数学计算，完成一个任务后再进行下一个任务。

下面的图表说明了循环神经网络的完整方法和工作方式−

PyTorch 循环神经网络

在上图中，c1、c2、c3和x1被视为包含一些隐藏输入值h1、h2和h3的输入，它们传递给o1的相应输出。我们现在将重点介绍使用PyTorch来实现递归神经网络创建正弦波的方法。

在训练过程中，我们将遵循一种一次处理一个数据点的训练方法来训练我们的模型。输入序列x包含20个数据点，目标序列被认为与输入序列相同。

步骤1

使用下面的代码导入实现递归神经网络所需的必要包。

import torch
from torch.autograd import Variable
import numpy as np
import pylab as pl
import torch.nn.init as init

步骤2

我们将设置模型的超参数，将输入层的大小设置为7。我们将有6个上下文神经元和1个输入神经元用于创建目标序列。

dtype = torch.FloatTensor
input_size, hidden_size, output_size = 7, 6, 1
epochs = 300
seq_length = 20
lr = 0.1
data_time_steps = np.linspace(2, 10, seq_length + 1)
data = np.sin(data_time_steps)
data.resize((seq_length + 1, 1))

x = Variable(torch.Tensor(data[:-1]).type(dtype), requires_grad=False)
y = Variable(torch.Tensor(data[1:]).type(dtype), requires_grad=False)

我们将生成训练数据，其中x是输入数据序列，y是所需的目标序列。

步骤3

使用均值为零的正态分布初始化递归神经网络中的权重。 W1将表示输入变量的接受度，w2将表示生成的输出，如下所示 –

w1 = torch.FloatTensor(input_size, 
hidden_size).type(dtype)
init.normal(w1, 0.0, 0.4)
w1 = Variable(w1, requires_grad = True)
w2 = torch.FloatTensor(hidden_size, output_size).type(dtype)
init.normal(w2, 0.0, 0.3)
w2 = Variable(w2, requires_grad = True)

步骤4

现在，重要的是要为前馈创建一个独特定义神经网络的函数。

def forward(input, context_state, w1, w2):
   xh = torch.cat((input, context_state), 1)
   context_state = torch.tanh(xh.mm(w1))
   out = context_state.mm(w2)
   return (out, context_state)

步骤5

下一步是开始循环神经网络正弦波实现的训练过程。外循环遍历每个循环，内循环遍历序列的每个元素。在这里，我们还会计算均方差（MSE），它有助于连续变量的预测。

for i in range(epochs):
   total_loss = 0
   context_state = Variable(torch.zeros((1, hidden_size)).type(dtype), requires_grad = True)
   for j in range(x.size(0)):
      input = x[j:(j+1)]
      target = y[j:(j+1)]
      (pred, context_state) = forward(input, context_state, w1, w2)
      loss = (pred - target).pow(2).sum()/2
      total_loss += loss
      loss.backward()
      w1.data -= lr * w1.grad.data
      w2.data -= lr * w2.grad.data
      w1.grad.data.zero_()
      w2.grad.data.zero_()
      context_state = Variable(context_state.data)
   if i % 10 == 0:
      print("Epoch: {} loss {}".format(i, total_loss.data[0]))

context_state = Variable(torch.zeros((1, hidden_size)).type(dtype), requires_grad = False)
predictions = []

for i in range(x.size(0)):
   input = x[i:i+1]
   (pred, context_state) = forward(input, context_state, w1, w2)
   context_state = context_state
   predictions.append(pred.data.numpy().ravel()[0])

步骤6

现在，是时候按照所需的方式绘制正弦波了。

pl.scatter(data_time_steps[:-1], x.data.numpy(), s = 90, label = "Actual")
pl.scatter(data_time_steps[1:], predictions, label = "Predicted")
pl.legend()
pl.show()

输出

上述过程的输出如下所示 –

PyTorch 循环神经网络