Research Article
Bus Single-Trip Time Prediction Based on Ensemble Learning
Table 7
Model training and prediction time.
| Models | TimeStep = 4 | TimeStep = 6 | Training | Prediction (ms) | Training | Prediction (ms) |
| LSTM | 85 min | 986 | 106 min | 1326 | GRU | 93 min | 645 | 114 min | 735 | LR | 4 ms | 1 | 6 ms | 2 | KNN | 64 ms | 52 | 71 ms | 68 | XGBoost | 2639 ms | 32 | 3283 ms | 42 | LR-stacking | 21 ms | 6 | 35 ms | 21 | RF-bagging | 214 ms | 51 | 257 ms | 82 | AdaBoost-boosting | 1427 ms | 72 | 1974 ms | 102 |
|
|