Optimized Machine Learning Model | Mean (SD) F1-score (train set, fivefold cross-validation) | ||
---|---|---|---|
3 months prediction | 6 months prediction | 9 months prediction | |
Logistic regression | 0.797 (0.012) | 0.785 (0.016) | 0.768 (0.012) |
Ordinal logistic regression | 0.798 (0.011) | 0.784 (0.013) | 0.768 (0.010) |
Random forest | 0.807 (0.012) | 0.798 (0.018) | 0.780 (0.011) |
Extra trees | 0.801 (0.017) | 0.783 (0.021) | 0.771 (0.01) |