[发明专利]一种列车到站晚点预测方法、装置、电子设备及存储介质

摘要

本发明公开了一种列车到站晚点预测方法、装置、电子设备及存储介质，该方法包括：提取列车晚点的特征集合，基于所述特征集合构建时空关联的XGBoost晚点预测模型；定义所述XGBoost晚点预测模型的超参数范围，将所述特征集合分割为训练集和测试集；基于所述训练集采用贝叶斯算法对所述XGBoost晚点预测模型的超参数组合进行迭代优化；采用最大化采集函数对所述优化后的XGBoost晚点预测模型进行修正；采用测试集测试所述修正后的XGBoost晚点预测模型以得到最终的预测结果。本发明提出的方法比既有的晚点预测模型预测精度更高、鲁棒性更强、计算速度更快，能够为晚点情况下列车能力评估及列车调整提供决策支持。