From 56c74ab5d3b9e1b43d6130e8438d54962ff2a259 Mon Sep 17 00:00:00 2001 From: Newnius Date: Sat, 2 May 2020 11:14: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 c43bd9f..eb6c4bd 100644 --- a/.idea/workspace.xml +++ b/.idea/workspace.xml @@ -226,7 +226,7 @@ - + diff --git a/train.py b/train.py index dfc06b9..b87163a 100644 --- a/train.py +++ b/train.py @@ -104,7 +104,7 @@ def experiment(repeats, series, seed): t2 = test.shape[0] % batch_size train_trimmed = train_scaled[train_scaled.shape[0] - t1 * batch_size:, :] - lstm_model = fit_lstm(train_trimmed, batch_size, 3000, 4) + lstm_model = fit_lstm(train_trimmed, batch_size, 50, 4) # forecast the entire training dataset to build up state for forecasting print(train_trimmed) print(train_trimmed[:, 0])