From 1cbe0bf37ba0993d0f22720ebd1b667dad18536e Mon Sep 17 00:00:00 2001 From: Newnius Date: Sat, 2 May 2020 10:55:56 +0800 Subject: [PATCH] update --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index fe383f9..8175ba5 100644 --- a/train.py +++ b/train.py @@ -108,9 +108,9 @@ def experiment(repeats, series, seed): # forecast the entire training dataset to build up state for forecasting print(train_trimmed) print(train_trimmed[:, 0]) - print(train_trimmed[:, :]) + print(train_trimmed[:, :-2]) if seed: - train_reshaped = train_trimmed[:, :].reshape(len(train_trimmed), 1, lag2+1) + train_reshaped = train_trimmed[:, :-2].reshape(len(train_trimmed), 1, lag2) lstm_model.predict(train_reshaped, batch_size=batch_size) # forecast test dataset test_reshaped = test_scaled[:, 0:-1]