pytorch-LSTM

最新推荐文章于 2024-09-18 12:21:11 发布

mkopvec

最新推荐文章于 2024-09-18 12:21:11 发布

阅读量520

点赞数

分类专栏： pytorch

本文链接：https://blog.csdn.net/mkopvec/article/details/90271768

版权

本文介绍如何使用PyTorch的LSTM模型来拟合周期性的sin函数，展示深度学习在非线性问题上的应用。

摘要由CSDN通过智能技术生成

练习使用LSTM,用它来实现拟合sin函数

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import math
dataset=[]
for data in np.arange(0, 3, .001):
    data = math.sin(data*math.pi)
    dataset.append(data)
dataset=np.array(dataset)
dataset = dataset.astype('float32')
max_value = np.max(dataset)
min_value = np.min(dataset)
scalar = max_value - min_value
dataset = list(map(lambda x: x / scalar, dataset))
def create_dataset(dataset, look_back=3):
    dataX, dataY = [],[]
    for i in range(len(dataset) - look_back):
        a = dataset[i:(i + look_back)]
        dataX.append(a)
        dataY.append(dataset[i + look_back])
    return np.array(dataX), np.array(dataY)
data_X, data_Y = create_dataset(dataset)
train_size = int(len(data_X) * 0.7)
test_size = len(data_X) - train_size
train_X = data_X[:train_size]
train_Y = data_Y[:train_size]
test_X = data_X[train_size:]
test_Y = data_Y