Abstract
BACKGROUND: Nurses' high workload can result in depressive symptoms. However, the research has underexplored the internal and external variables, such as organisational support, career identity, and burnout, which may predict depressive symptoms among Chinese nurses via machine learning (ML). AIM: To predict nurses' depressive symptoms and identify the relevant factors by machine learning (ML) algorithms. METHODS: A self-administered smartphone questionnaire was delivered to nurses to evaluate their depressive symptoms; 1,431 questionnaires and 28 internal and external features were collected. In the training set, the use of maximum relevance minimum redundancy ranked the features' importance. Five ML algorithms were used to establish models to identify nurses' depressive symptoms using different feature subsets, and the area under the curve (AUC) determined the optimal feature subset. Demographic characteristics were added to the optimal feature subset to establish the combined models. Each model's performance was evaluated using the test set. RESULTS: The prevalence rate of depressive symptoms among Chinese nurses was 31.86%. The optimal feature subset comprised of sleep disturbance, chronic fatigue, physical fatigue, exhaustion, and perceived organisation support. The five models based on the optimal feature subset had good prediction performance on the test set (AUC: 0.871-0.895 and accuracy: 0.798-0.815). After adding the significant demographic characteristics, the performance of the five combined models slightly improved; the AUC and accuracy increased to 0.904 and 0.826 on the test set, respectively. The logistic regression analysis results showed the best and most stable performance while the univariate analysis results showed that external and internal personal features (AUC: 0.739-0.841) were more effective than demographic characteristics (AUC: 0.572-0.588) for predicting nurses' depressive symptoms. CONCLUSIONS: ML could effectively predict nurses' depressive symptoms. Interventions to manage physical fatigue, sleep disorders, burnout, and organisational support may prevent depressive symptoms.