|
Abstract
|
OBJECTIVE To explore the predictive effect of machine learning algorithms on college students' suicidal ideation and to analyze the associated factors of college students' suicidal ideation.
METHODS The mental health data of 21 224 undergraduates was selected from a university in 2021. The independent variables were 37 demographic and internal and external mental health factors. The dependent variable was whether college students had suicidal ideation. Support vector machine, random forest and LightG-BM algorithm were used to establish prediction models. The model was used in test set to so as to evaluate the model's prediction effect by using detection rate, F1 score and accuracy rate. Based on the superior model, the high-risk factors of suicidal ideation in college students were analyzed.
RESULTS The detection rates of support vector machine, random forest, and LightGBM models were 61.0%, 64.0%. 69.0%; F1 scores were 0.63, 0.63. 0.64, and accuracy rates were 73.0%, 73.0%, 72.0%, respectively. Based on the superior Light GBM model, risk factors of suicidal ideation in college students included, depression, grade, gender, despair, place of origin, sense of meaning, attitude toward suicide, dependence, family economic situation, hallucinatory delusion symptoms, anxiety, internet addiction, and interpersonal distress.
CONCLUSION The LightGBM model has a better prediction effect than the support vector machine and random forest models. © Journal Office of Chinese School Health(Bengbu). All rights reserved.
Language: zh |