Zhaoliang Meng, Zetao Zhang, Yuan Yang, Guofeng Li, Chongbo Tao, Yijiang Niu. Improved XGBoost Stray Current Prediction and Interpretable Model[J]. Laser & Optoelectronics Progress, 2022, 59(12): 1215011
- Laser & Optoelectronics Progress
- Vol. 59, Issue 12, 1215011 (2022)
Fig. 1. Error comparision under different characteristic number combinations
Fig. 2. Feature combination error contrast histograms. (a) MAE; (b) MSE; (c) RMSE; (d) R2
Fig. 3. Optimal characteristic iterative procession
Fig. 4. Comparison of different models for stray current prediction
Fig. 5. SHAP feature analysis sketch chart
Fig. 6. Dependence diagram of rail voltage SHAP analysis
Fig. 7. Dependence diagram of transition resistance SHAP analysis
|
Table 1. Error comparison between original model and optimized model
|
Table 2. Prediction performance index of defferent models based on optimal feature combination method
Set citation alerts for the article
Please enter your email address
© Copyright 2018-2021 | Chinese Laser Press. All Rights Reserved 沪ICP备15018463号-20