Research and Application of Deformation Prediction Model for Deep Foundation Pit Based on LSTM

Algorithm 2

Momentum algorithm.

Require: set the initial learning rate to , set to momentum coefficient

Require: set the initial parameter to , set initial speed to 0

While stop condition not satisfied do

Randomly extract samples from the training set, and is the true value corresponding to .

Calculate the average gradient of samples:

Step size update:

Parameter updating:

End while