From edf61d117d394a84b9f03b6770e1198d7b31ba43 Mon Sep 17 00:00:00 2001 From: Newnius Date: Sat, 2 May 2020 09:52:48 +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 fc5d277..37472b1 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -224,7 +224,7 @@ - + diff --git a/train.py b/train.py index b46565e..9f63cf8 100644 --- a/train.py +++ b/train.py @@ -104,7 +104,7 @@ train, test = supervised_values[0:-12], supervised_values[-12:] scaler, train_scaled, test_scaled = scale(train, test) # fit the model -lstm_model = fit_lstm(train_scaled, 1, 300, 4) +lstm_model = fit_lstm(train_scaled, 1, 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)