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])