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 |