diff --git a/train.py b/train.py index f09f7b5..b09f6b8 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, 3000, 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)