Supplementary file 3_Development and validation of a machine learning-based return to work predictive model for young and middle-aged maintenance hemodialysis patients.docx

The optimal algorithm was selected based on comprehensive evaluation metrics, including area under curve (AUC), calibration curves, and net benefit from decision curve analysis. Least absolute shrinkage and selection operator (LASSO) regression was subsequently applied for feature selection. After determining the final feature set, the optimal model was retrained on the modeling set.

Internal validation was conducted using nested cross-validation, and external validation was performed on an independent validation set. The final model was explained using SHapley Additive exPlanations and visualized using a web-based calculator.</p>Results<p>The RF model was identified as the optimal predictive model. LASSO regression further selected 10 feature variables, including RTW self-efficacy, social functioning, chronic comorbidities, social support, social isolation, family functioning, age, dialysis time period, residence and gender.

The RF model was retrained using the selected features. The internal validation yielded an AUC of 0.946 (95% CI 0.892–0.997), while the external validation achieved an AUC of 0.918 (0.873–0.963), indicating good predictive performance and generalizability of the model.</p>Conclusion<p>The RF model performs well in predicting current RTW probability among young and middle-aged MHD patients, enabling identification of high-risk non-RTW individuals and key contributing factors.

It may serve as a practical screening tool for precision rehabilitation and stratified clinical management.</p>