Skip to main content

Table 4 Performance of prognostic models built by machine learning algorithms on test data (area under the ROC curve)

From: Novel models by machine learning to predict prognosis of breast cancer brain metastases

 

6-month survival

1-year survival

2-year survival

3-year survival

XGBoost

0.824

0.813

0.800

0.803

LR

0.794

0.744

0.740

0.744

RF

0.770

0.729

0.730

0.756

SVM

0.730

0.647

0.525

0.509

KNN

0.738

0.623

0.581

0.586

ID3

0.692

0.628

0.685

0.639

  1. XGBoost extreme gradient boosting, LR logistic regression, RF random forest, SVM support vector machine, ID3 decision tree, KNN K-Nearest Neighbor
Back to article page

Journal of Translational Medicine

ISSN: 1479-5876

Contact us