Research Article

Unidirectional and Bidirectional LSTM Models for Short-Term Traffic Prediction

Table 1

Model hyperparameters for Uni-LSTM and BiLSTM.

Gradient decay factor0.9
Initial learning rate0.005
Minimum batch size128
Maximum epochs300
Training optimizerAdaptive moment estimation optimizer
Dropping learning rate during trainingPiecewise
Learning rate drop period125
Factor for learning rate dropping0.2