训练 LSTM 的模型和使用模型去预测

LSTM  模型的训练过程：

LSTM  模型运行的结果： 

#Code borrowed from: 骆旺达
#Modify by tony2278  20200409

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import torch

data_csv = pd.read_csv('./data.csv',usecols=[1])
#plt.plot(data_csv)
#plt.show()

# 数据预处理
data_csv = data_csv.dropna()  # 滤除缺失数据
dataset = data_csv.values   # 获得csv的值
dataset = dataset.ast