Hybrid data-driven model for predicting the peak shear strength of rock joints
摘要
Accurately predicting the peak shear strength (PSS) of rock joints is of great significance for rock engineering and mining engineering. In this study, the whale optimization algorithm (WOA) and the sparrow search algorithm (SSA) were used to optimize the hyperparameters of the extreme gradient boosting (XGBoost) model, and two hybrid models, WOA-XGBoost and SSA-XGBoost, were established for predicting PSS. The performance of these two hybrid models was compared with XGBoost, gene expression programming (GEP), group method of data handling (GMDH), and two empirical models. The results show that the hybrid SSA-XGBoost model has the highest prediction accuracy, with its coefficient of determination (R2), root mean squared error (RMSE), mean absolute error (MAE), Willmott's Index of Agreement (WIA), and Legates-McCabe's Index (LM) being 0.9906, 0.2861 MPa, 0.1312 MPa, 0.9994, and 0.9591, respectively. Additionally, the SHapley additive explanations (SHAP) method was used to analyze the importance of each parameter on PSS.