模型类别 | 主要参数 | 取值 |
支持向量机 | 核函数 | 高斯核函数 |
错误项的惩罚系数 | 1.0 | |
阈值 | 0.01 | |
最大迭代次数 | 1000 | |
随机森林 | 决策树棵树 | 200 |
特征数 | 11 | |
最大深度 | 10 | |
结点最小样本树 | 15 | |
XGBoost | 学习率 | 0.2 |
最小权重 | 1 | |
树深度 | 8 | |
决策树棵树 | 100 | |
BP神经网络 | 学习率 | 0.01 |
卷积核个数 | 32 | |
卷积核大小 | 5 × 5 | |
池化核大小 | 2 × 2 | |
激活函数 | ReLU |