From d500ec5877ebfffd78c8555ce4eae840713b160e Mon Sep 17 00:00:00 2001 From: Newnius Date: Sat, 2 May 2020 09:28:09 +0800 Subject: [PATCH] update --- .idea/workspace.xml | 2 +- train.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.idea/workspace.xml b/.idea/workspace.xml index eb21a92..199c7fb 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -223,7 +223,7 @@ - + diff --git a/train.py b/train.py index 8b5b777..7a64ced 100644 --- a/train.py +++ b/train.py @@ -102,7 +102,7 @@ scaler, train_scaled, test_scaled = scale(train, test) lstm_model = fit_lstm(train_scaled, 32, 30, 4) # forecast the entire training dataset to build up state for forecasting train_reshaped = train_scaled[:, 0].reshape(len(train_scaled), 1, 1) -lstm_model.predict(train_reshaped, batch_size=1) +lstm_model.predict(train_reshaped, batch_size=32) # walk-forward validation on the test data predictions = list()