Table 5 AUC-ROC and accuracy for SVM with the linear and RBF kernels as the best models trained with miRNAs selected based on their AUC-ROC on the train and test sets before and after hypertuning
Model | Metrics | Preset parameters | Hypertuned | ||
|---|---|---|---|---|---|
Train | Test | Train | Test | ||
SVM-linear | AUC-ROC | 0.91 | 0.93 | 0.93 | 0.92 |
Accuracy | 0.85 | 0.82 | 0.90 | 0.88 | |
SVM-RBF | AUC-ROC | 0.90 | 0.92 | 0.96 | 0.96 |
Accuracy | 0.86 | 0.84 | 0.96 | 0.94 | |