Skip to main content

Table 2 Predictive performance of the six machine learning models

From: Prediction of 5-year overall survival of tongue cancer based machine learning

Models(18 features)

AUC

Accuracy

Precision

Recall

F1

Logistic Regression

0.845

0.615

0.554

1

0.713

SVC

0.831

0.738

0.719

0.742

0.730

Decision Tree

0.712

0.708

0.688

0.710

0.698

RF

0.850

0.785

0.743

0.839

0.788

XGB

0.790

0.692

0.649

0.774

0.706

LGBM

0.790

0.692

0.649

0.774

0.706

  1. Abbreviations: SVC Support vector machines, RF Random Forest, XGB eXtreme Gradient Boosting(XGB), LGBM Light Gradient Boosting Machine, AUC Area Under the Curve