Skip to main content
. 2019 Feb 20;14(2):e0210976. doi: 10.1371/journal.pone.0210976

Table 2. Performance of machine learning models for predicting recurrence and survival.

Post-cystectomy outcome Model Year Test set performance metrics
Sensitivity Specificity Precision F1
Recurrence Meta-classifier 1 0.739 0.714 0.388 0.508
pT stage TNM 5th Edition 1 0.761 0.653 0.349 0.478
pathologic stage subgroup 1 0.826 0.593 0.332 0.473
Meta-classifier 3 0.720 0.708 0.535 0.613
pathologic stage subgroup 3 0.774 0.631 0.493 0.602
pT stage TNM 5th Edition 3 0.670 0.694 0.503 0.574
Meta-classifier 5 0.700 0.702 0.588 0.636
pathologic stage subgroup 5 0.744 0.611 0.537 0.623
pT stage TNM 5th Edition 5 0.619 0.698 0.553 0.583
Survival Meta-classifier 1 0.741 0.770 0.473 0.577
pT stage TNM 5th Edition 1 0.739 0.672 0.387 0.506
pathologic stage subgroup 1 0.805 0.602 0.362 0.499
Meta-classifier 3 0.722 0.788 0.700 0.711
pathologic stage subgroup 3 0.762 0.691 0.628 0.688
pT stage TNM 5th Edition 3 0.696 0.739 0.646 0.670
Meta-classifier 5 0.741 0.768 0.780 0.760
pathologic stage subgroup 5 0.730 0.717 0.742 0.735
pT stage TNM 5th Edition 5 0.664 0.766 0.759 0.708

Single predictor (pT stage and pathologic stage subgroup classifiers) and multiple predictor (Meta-classifier) models for predicting 1-, 3-, 5-year recurrence and survival after cystectomy. The performance of all models for a given year is ranked per F1 scores (2*precision*recall/(precision+recall)) as well as mean sensitivity, specificity, and precision on test sets from a 10-fold cross validation.