From 6f26c20c4329ee8a9dfdc9605861b6f9414b488f Mon Sep 17 00:00:00 2001 From: Newnius Date: Sat, 2 May 2020 10:56:50 +0800 Subject: [PATCH] update --- .idea/workspace.xml | 19 +++++++++---------- train.py | 4 ++-- 2 files changed, 11 insertions(+), 12 deletions(-) diff --git a/.idea/workspace.xml b/.idea/workspace.xml index 9870cd7..b361b19 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -2,7 +2,6 @@ - @@ -48,7 +47,7 @@ - + @@ -57,7 +56,7 @@ - + @@ -92,8 +91,8 @@ - - + + @@ -226,7 +225,7 @@ - + @@ -267,12 +266,12 @@ - @@ -397,8 +396,8 @@ - - + + diff --git a/train.py b/train.py index 8175ba5..be0f209 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[:, :-2]) + print(train_trimmed[:, :-1]) if seed: - train_reshaped = train_trimmed[:, :-2].reshape(len(train_trimmed), 1, lag2) + train_reshaped = train_trimmed[:, :-1].reshape(len(train_trimmed), 1, lag2) lstm_model.predict(train_reshaped, batch_size=batch_size) # forecast test dataset test_reshaped = test_scaled[:, 0:-1]