基于小批量梯度下 降法的预测模型

batch size

= 64

(α = 0.0015)

batch size

= 128

(α = 0.0013)

batch size

= 512

(α = 0.0016)

batch size

= 1024

(α = 0.0018)

batch size

= 2048

(α = 0.0001)

RMSE

0.930

0.913

0.909

0.903

0.941

MAE

0.742

0.725

0.718

0.714

0.753

Time

182.55secs

90.83 secs

48.74 secs

20.41 secs

14.33 secs