From 3c580a33d3521a08149162e311b4212dd4ed188b Mon Sep 17 00:00:00 2001 From: Newnius Date: Sat, 2 May 2020 09:48: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 a383f35..fc5d277 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -224,7 +224,7 @@ - + diff --git a/train.py b/train.py index 9f63cf8..b46565e 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, 30, 4) +lstm_model = fit_lstm(train_scaled, 1, 300, 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)