基于LightGBM算法的公交行程时间预测

Bus Travel Time Prediction Based on Light Gradient Boosting Machine Algorithm

在城市公交网络运行中,公交车的站点间行程时间会受到道路和环境条件的影响.本文对公交车运行过程中的车辆速度特征、道路特征及天气特征等进行了分析.建立了基于特征的LightGBM (Light Gradient Boosting Machine)公交行程时间预测模型,通过调整LightGBM算法中的相关参数,以分配各个影响特征和因素的权重大小.然后利用天津市某条公交线路24天的公交车GPS数据对模型进行了训练和验证,并与基于历史平均值和卡尔曼滤波的行程时间预测模型进行对比.比较结果表明,LightGBM模型在MAE (Mean Absolute Error)和MAPE (Mean Absolute Percentage Error)这两个指标上均大幅度优于其他两个模型,说明LightGBM模型在公交车行程时间预测上具有很好的稳定性和应用前景.

In the operation of urban public transport networks, the travel time between stations of a bus is affected by road and environmental conditions. This paper analyzes the bus speed characteristics, road characteristics and weather characteristics during bus operation, and a feature-based LightGBM bus travel time prediction model is established. By adjusting the relevant parameters in the LightGBM algorithm, the weights of the influencing features and factors are assigned. Then the model is trained and verified by using the 24-day bus GPS data of one bus line in Tianjin, and compared with the travel time prediction model based on historical mean and Kalman filter. The comparison results show that the LightGBM model is superior to the other two models in MAE (Mean Absolute Error) and MAPE (Mean Absolute Percentage Error), indicating that the LightGBM model has good stability and application prospects in bus travel time prediction.