基于可解釋機(jī)器學(xué)習(xí)框架的列車乘車區(qū)段客流分布預(yù)測方法

摘要： 為解釋客運(yùn)產(chǎn)品特征對列車乘車區(qū)段客流分布預(yù)測的影響，本文提出一種基于可解釋機(jī)器學(xué)習(xí)框架的高速鐵路列車乘車區(qū)段客流分布預(yù)測方法。首先，提出基于梯度提升樹模型的高速鐵路列車乘車區(qū)段客流分布預(yù)測框架，構(gòu)建不同梯度提升樹模型(GBDT、XGBoost、LightGBM及CatBoost)的高速鐵路列車乘車區(qū)段客流分布預(yù)測模型；其次，計算特征貢獻(xiàn)重要度，基于SHAP(SHapley Ad...