基于XGBoost-SHAP的钢管混凝土柱轴向承载力预测模型.pdfVIP

第 57 卷第 6 期 浙    江    大    学    学    报 (工学版) Vol.57  No.6    2023 年 6 月 Journal of Zhejiang University (Engineering Science) Jun.  2023     DOI: 10.3785/j.issn.1008-973X.2023.06.001 基于XGBoost-SHAP 的钢管混凝土柱轴向 承载力预测模型 陈曦泽 ，贾俊峰 ，白玉磊 ，郭彤 ，杜修力1 1 1 2 1 (1. 北京工业大学 城市建设学部，北京 100124；2. 东南大学 土木工程学院，江苏 南京 211189) 摘  要：为了可靠、准确地预测钢管混凝土（CFST）柱的轴向承载力，建立和解释集成机器学习的CFST柱轴向承 载力预测模型. 使用马氏距离评估CFST柱数据库质量，通过极限梯度提升(XGBoost)算法建立CFST柱轴向承载 力预测模型，使用K 折交叉验证(K-Fold CV)和树结构概率密度估计(TPE)算法寻找模型的最优超参数组合. 采用 不同评价指标将优化后XGBoost模型的预测值与已有方法和未优化XGBoost模型的计算值比较. 使用SHAP方法 给出XGBoost模型预测结果的整体和局部的解释. 结果表明，经过超参数调整优化的XGBoost模型的性能超越了 相关规范和经验公式的性能，且SHAP方法能够有效地解释XGBoost模型的输出. 关键词： 钢管混凝土（CFST）柱；轴向承载力；极限梯度提升（XGBoost）；超参数优化；SHAP；可解释性 中图分类号： TU 375           文献标志码： A           文章编号： 1008−973X（2023）06−1061−10 Prediction model of axial bearing capacity of concrete-filled steel tube columns based on XGBoost-SHAP 1 1 1 2 1 CHEN Xi-ze,  JIA Jun-feng,  BAI Yu-lei,  GUO Tong,  DU Xiu-li (1. Faculty of Architecture, Civil and Transportation Engineering, Beijing University of Technology, Beijing 100124, China; 2. School of Civil Engineering, Southeast University, Nanjing 211189, China) Abstract: To reliably and accurately predict the axial bearing capacity of concrete-filled steel tube (CFST) columns, a prediction model of CFST column axial b