pISSN 1225-388X
eISSN 2093-7784
Grid search for the optimal hyperparameters
| Model | ntree | mtry | AUC | CA | F | Precision | Recall |
|---|---|---|---|---|---|---|---|
| Tree | 0.718 | 0.724 | 0.699 | 0.696 | 0.724 | ||
| Random forest | 100 | 3 | 0.736 | 0.729 | 0.703 | 0.697 | 0.719 |
| Random forest | 50 | 3 | 0.736 | 0.720 | 0.703 | 0.698 | 0.720 |
| Random forest | 100 | 4 | 0.737 | 0.721 | 0.705 | 0.700 | 0.721 |
| Random forest | 50 | 4 | 0.736 | 0.720 | 0.705 | 0.699 | 0.720 |
| Random forest | 100 | 5 | 0.736 | 0.720 | 0.703 | 0.698 | 0.720 |
| Random forest | 50 | 5 | 0.735 | 0.721 | 0.705 | 0.699 | 0.721 |