Table 2 Evaluation of mRNA machine learning models.

From: Development and validation of machine learning models for diagnosis and prognosis of lung adenocarcinoma, and immune infiltration analysis

Learner

auc_train

auc_test

prauc_train

prauc_test

classif.kknn

1.0000000

1.0000000

1.0000000

1.0000000

classif.naive_bayes

1.0000000

0.9973684

1.0000000

0.9997300

classif.ranger

1.0000000

1.0000000

1.0000000

1.0000000

classif.rpart

0.9844472

0.9337719

0.9942892

0.9881881

classif.svm

1.0000000

1.0000000

1.0000000

1.0000000

classif.xgboost

0.9979483

0.9364035

0.9996251

0.9884611

  1. PRAUC, the area under the precision-recall curve; AUC, the area under the ROC curve.
Back to article page