LSTM单一维度预测

# 导入资源
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
from sklearn.preprocessing import MinMaxScaler
from keras.models import Sequential
from keras.layers import Dense
from keras.layers import LSTM

# 设置plt中文字体
plt.rcParams['font.family'] = ['sans-serif']
plt.rcParams['font.sans-serif'] = ['SimHei']


class SingleDimensionPrediction:

    def __init__(self, csv_path, loc):
        # 导入训练数据集
        csv_data = pd.read_csv(csv_path)
        # training_set = training_set.iloc[:, 1:2].values
        csv_data = csv_data[[loc]].values

        self.sc = MinMaxScaler()
        csv_data = self.sc.fit_transform(csv_data)

        dataX = csv_data[0: len(csv_data) - 1]
        dataY = csv_data[1: len(csv_data)]

        csv_data_len = len(dataX)
        train_len = int(csv_data_len * 80 / 100)
        test_len = csv_data_len - train_len

        self.X_train = dataX[:train_len]