Table 2

Performance of machine learning models in predicting treatment response

ModelRisperidoneAripiprazole
BSAUROCAUPRCBSAUROCAUPRC
Backward stepwise LR0.1240.6590.2440.1170.6720.300
LASSO0.1220.6780.2530.1150.6810.304
Ridge0.1230.6800.2550.1170.6830.307
e-net0.1220.6760.2520.1150.6830.305
Random forest0.1240.6580.2500.1150.6780.317
GBM0.1210.6860.2690.1140.6820.314
SuperLearner0.1210.6850.2640.1150.6880.314
  • AUPRC, area under the precision-recall curve; AUROC, area under the receiver operating characteristic curve; BS, Brier score; e-net, elastic net; GBM, gradient boosting machine; LASSO, least absolute shrinkage and selection operator; LR, logistic regression.