Research Article
Bus Single-Trip Time Prediction Based on Ensemble Learning
Table 3
Model training and prediction time.
| Models | TimeStep = 4 | TimeStep = 6 | Training | Prediction (ms) | Training | Prediction (ms) |
| LSTM | 115 min | 1257 | 121 min | 1471 | GRU | 119 min | 727 | 126 min | 1103 | LR | 7 ms | 2 | 16 ms | 3 | KNN | 77 ms | 63 | 98 ms | 82 | XGBoost | 3619 ms | 48 | 5192 ms | 73 | LR-stacking | 58 ms | 11 | 52 ms | 56 | RF-bagging | 329 ms | 79 | 417 ms | 136 | AdaBoost-boosting | 2351 ms | 92 | 3162 ms | 141 |
|
|