期刊
PSYCHIATRY INVESTIGATION
卷 16, 期 8, 页码 588-593出版社
KOREAN NEUROPSYCHIATRIC ASSOC
DOI: 10.30773/pi.2019.06.19
关键词
Suicide attempt; Suicide ideation; Machine learning; Public health data
类别
资金
- Korea Health Technology R&D Project through the Korea Health Industry Development Institute (KHIDI) - Ministry of Health & Welfare, Republic of Korea [HI17C0682]
Objective We aimed to develop predictive models to identify suicide attempters among individuals with suicide ideation using a machine learning algorithm. Methods Among 35,116 individuals aged over 19 years from the Korea National Health & Nutrition Examination Survey, we selected 5,773 subjects who reported experiencing suicide ideation and had answered a survey question about suicide attempts. Then, we performed resampling with the Synthetic Minority Over-sampling TEchnique (SMOTE) to obtain data corresponding to 1,324 suicide attempters and 1,330 non-suicide attempters. We randomly assigned the samples to a training set (n=1,858) and a test set (n=796). In the training set, random forest models were trained with features selected through recursive feature elimination with 10-fold cross validation. Subsequently, the fitted model was used to predict suicide attempters in the test set. Results In the test set, the prediction model achieved very good performance [area under receiver operating characteristic curve (AUC):=0.947] with an accuracy of 88.9%. Conclusion Our results suggest that a machine learning approach can enable the prediction of individuals at high risk of suicide through the integrated analysis of various suicide risk factors.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据