LSTMの予想ではある時点からいくつか前のデータを用いて
次の時点のデータを予測し教師データを作成する
入力データがある地点からいくつか前の点のデータ
教師データは次の時点のデータ となるように
入力データ、教師データを作成
次の関数のlook_back=nのnが「いくつ前のデータを利用するか」を設定
|
1 2 3 4 5 6 7 8 9 10 11 |
def create_dataset(dataset, look_back): dataX, dataY = [], [] for i in range(len(dataset)-look_back-1): a = dataset[i:(i+look_back), 0] dataX.append(a) dataY.append(dataset[i + look_back, 0]) return numpy.array(dataX), numpy.array(dataY) look_back = 10 trainX, trainY = create_dataset(train, look_back) testX, testY = create_dataset(test, look_back) |
コメントを残す