From c9ff63fbbf2710b44a5070c7f7468d0a3a97f354 Mon Sep 17 00:00:00 2001 From: Newnius Date: Sat, 2 May 2020 15:08:12 +0800 Subject: [PATCH] update --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 448f648..ea3a88f 100644 --- a/train.py +++ b/train.py @@ -108,7 +108,7 @@ def experiment(repeats, series, seed): t2 = test.shape[0] % batch_size train_trimmed = train_scaled[t1:, :] - lstm_model = fit_lstm(train_trimmed, batch_size, 300, 4) + lstm_model = fit_lstm(train_trimmed, batch_size, 30, 4) # forecast the entire training dataset to build up state for forecasting print(train_trimmed) print(train_trimmed[:, 0])