Table 1 Finalised parameters for each model.
Model | Parameter | PSM | CM |
|---|---|---|---|
ANN | No. of layers | 3 | 6 |
Neurons per layer | 10 | 15 | |
Activation function | tansig | tansig | |
Training algorithm | train lm | train lm | |
Regularization | 0.01 | 1.01 | |
XGB | Learing rate | 0.1 | 0.3 |
max_depth | 5 | 7 | |
min_child_weight | 1 | 5 | |
n_estimators | 100 | 100 | |
subsample | 0.8 | 1 | |
RF | numTrees | 50 | 50 |
maxFeatures | 5 | 3 | |
minLeafSize | 5 | 5 |
