基于集成树算法的岩石黏聚力和内摩擦角预测方法

基于集成树算法的岩石黏聚力和内摩擦角预测方法

李地元,杨博,刘子达,刘永平,赵君杰

Prediction Method of Rock Cohesion and Internal Friction Angle Based on Ensemble Tree Algorithm

Diyuan LI,Bo YANG,Zida LIU,Yongping LIU,Junjie ZHAO

表2 模型的超参数优化范围

Table 2 Hyperparameter optimization range of the model

模型	超参数	优化范围
RF	回归树数量	（10，100）
RF	树的最大深度	（1，10）
ET	回归树数量	（10，100）
ET	树的最大深度	（1，10）
AdaBoost	提升迭代次数	（10，100）
AdaBoost	学习率	（0.01，0.20）
GBDT	提升迭代次数	（10，100）
	学习率	（0.01，0.20）
	树的最大深度	（1，10）
CatBoost	提升迭代次数	（100，1 200）
CatBoost	学习率	（0.01，0.20）