Table 2 The details of hyper-parameters for ML algorithm
Num | Algorithms | Hyper-parameters |
|---|---|---|
1 | SVM | kernel=’rbf’, degree = 3, C = 1.0, probability = True |
2 | Random Forest | n_estimators = 10, max_depth = None, min_samples_split = 2 |
3 | Extra Trees | n_estimators = 10, max_depth = None, min_samples_split = 2 |
4 | XGBoost | n_estimators = 10, max_depth = 5 |
5 | LightGBM | n_estimators = 10, max_depth = 4 |