RNN实现回归模型

最新推荐文章于 2024-10-27 16:22:58 发布

篱下浅歌生

最新推荐文章于 2024-10-27 16:22:58 发布

阅读量891

点赞数

分类专栏：莫凡pytorch 文章标签： rnn 回归深度学习

本文链接：https://blog.youkuaiyun.com/weixin_42318554/article/details/122510662

版权

文章目录

RNN实现回归模型
一、代码

RNN实现回归模型

用 sin 的曲线预测出 cos 的曲线.

一、代码

"""
View more, visit my tutorial page: https://mofanpy.com/tutorials/
My Youtube Channel: https://www.youtube.com/user/MorvanZhou
Dependencies:
torch: 0.4
matplotlib
numpy
"""
import torch
from torch import nn
import numpy as np
import matplotlib.pyplot as plt

# torch.manual_seed(1)    # reproducible

# Hyper Parameters
TIME_STEP = 10  # rnn time step
INPUT_SIZE = 1  # rnn input size
LR = 0.02  # learning rate

# show data
steps = np.linspace(0, np.pi * 2, 100, dtype=np.float32)  # float32 for converting torch FloatTensor
x_np = np.sin(steps)
y_np = np.cos(steps)
plt.plot(steps, y_np, 'r-', label='target (cos)')
plt.plot(steps, x_np, 'b-', label='input (sin)')
plt.legend(loc=