import torch
from torch import nn
import numpy as np
import matplotlib as mat
mat.use("TkAgg")
import matplotlib.pyplot as plt
import time
from torch.autograd import Variable
import cv2
torch.manual_seed(1) # reproducible
LR = 0.02 # learning rate
class LSNN(nn.Module):
def __init__(self):
super(LSNN, self).__init__()
self.lstm = nn.LSTM(
input_size=1,
hidden_size=32,
num_layers=1,
batch_first=True, # input & output 会是以 batch size 为第一维度的特征集 e.g. (batch, time_step, input_size)
)
self.hidden = (torch.autograd.Variable(torch.zeros(1, 1, 32)),torch.autograd.Variable(torch.zeros(1, 1, 32)))
self.out = nn.Linear(32, 1)
def forward(self,x):
# x (batch, time_step, input_size)
# h_state (n_layers, batch, hidden_size)
# r_out (batch, time_step, output_size)
r_out,self.hidden= self.l
基于pytorch使用LSTM神经网络预测时序模型
最新推荐文章于 2024-08-13 15:48:31 发布
本文详细介绍了如何使用PyTorch库构建并训练LSTM神经网络,针对时序数据进行预测。通过实例展示从数据预处理到模型构建、训练、验证和预测的全过程,揭示LSTM在网络中捕捉长期依赖性的能力。
摘要由CSDN通过智能技术生成