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)