一种利用贝叶斯优化的蓝藻遥感分类方法

田晨, 张金龙, 金义蓉, 董世元, 王彬, 张乃祥

A remote sensing classification method for cyanobacteria using Bayesian optimization algorithm

TIAN Chen, ZHANG Jinlong, JIN Yirong, DONG Shiyuan, WANG Bin, ZHANG Naixiang

表2 XGBoost最优超参数组合

Tab.2 XGBoost optimal hyperparameter combination

变量 含义 取值范围 步长 最优值 max_depth 子模型最大深度 [5, 15] 1 4 learning_rate 学习率 0.001, 0.01, 0.1, 1, 10 — 0.1 n_estimators 子模型数量 [10, 300] 10 35 subsample 随机采样的比例 (0, 1) 0.1 0.5 colsample_bytree 特征占比 (0, 1] 0.1 0.4 min_child_weight 最小叶节点样本权重和 [1, 10] 1 4