Table.
Model Performance (Test Performance, I.E., Outer-Loop of the Nested Cross Validation)
| Model | Mean Absolute Error (Injections/Year) [95% CI] | Cross-Validated R2 | ROC-AUC for Low Treatment Requirement* | ROC-AUC for High Treatment Requirement † |
|---|---|---|---|---|
| LASSO regression | 2.76 [2.39–3.14] | 0.038 | 0.61 | 0.63 |
| Principal component regression | 2.74 [2.38–3.11] | 0.173 | 0.63 | 0.7 |
| Random forest regression | 2.60 [2.25–2.96] | 0.390 | 0.68 | 0.7 |
| NGBoost | 2.66 [2.31–3.01] | 0.094 | 0.68 | 0.69 |