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中文字体
from HDFS import HDFS

plt.rcParams['font.family'] = ['sans-serif']
plt.rcParams['font.sans-serif'] = ['SimHei']


class MultiDimensionPrediction:

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

        self.sc = MinMaxScaler()
        csv_data_train = self.sc.fit_transform(csv_data_train)
        csv_data_test = self.sc.fit_transform(csv_data_test)

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