Research Article

Prediction and Impact Analysis of Passenger Flow in Urban Rail Transit in the Postpandemic Era

Table 2

Parameter settings of the GRU neural network.

ParametersDescriptionValues

epochNumber of iterations100
time_stepNumber of timesteps8
input_sizeDimensions of the input feature vector6
hidden_sizeDimensions of the hidden feature vector64
num_layersNumber of hidden layers1
lrLearning rate0.02
optimizerType of optimizerAdam
weight_decayCoefficient of weight decay0.001
loss_funcType of loss functionMSELoss