Journal of Human Earth and Future, Volume 5, Issue 1, Pages 1-18 , 01/03/2024

An Analysis of Machine Learning for Detecting Depression, Anxiety, and Stress of Recovered COVID-19 Patients

Tran Anh Tuan, Le Thanh Thao Trang, Tran Dai An, Nguyen Huu Nghia, Dao Thi Thanh Loan

Abstract

Objectives: This study explores different machine learning models (KNN: k-nearest neighbor, MLP: Multilayer Perceptron, SVM: Support Vector Machine) to identify the optimal model for accurate and rapid mental health detection among the recovered COVID-19 patients. Other techniques are also investigated, such as feature selection (Recursive Feature Elimination (RFE) and Extra Trees (ET) methods) and hyper-parameter tuning, to achieve a system that could effectively and quickly indicate mental health. Method/Analysis: To achieve the objectives, the study employs a dataset collected from recovered COVID-19 patients, encompassing information related to depression, anxiety, and stress. Machine learning models are utilized in the analysis. Additionally, feature selection methods and hyper-parameter tuning techniques are explored to enhance the model’s predictive capabilities. The performance of each model is assessed based on accuracy metrics. Findings: The experimental results show that SVM is the most suitable model for accurately predicting an individual’s mental health among recovered COVID-19 patients (accuracy ≥ 0.984). Furthermore, the ET method is more effective than the RFE method for feature selection in the anxiety and stress datasets. Novelty/Improvement: The study lies in the understanding of predictive modeling for mental health and provides insights into the choice of models and techniques for accurate and early detection.

Document Type

Article

Source Type

Journal

Keywords

AnxietyCOVID-19DASDepressionMachine LearningMental HealthPredictive ModelStress

ASJC Subject Area

Agricultural and Biological Sciences : Agricultural and Biological Sciences (miscellaneous)Engineering : Engineering (miscellaneous)Environmental Science : Environmental Science (miscellaneous)

Funding Agency

Walailak University


Bibliography


Tuan, T., Trang, L., An, T., Nghia, N., & Loan, D. (2024). An Analysis of Machine Learning for Detecting Depression, Anxiety, and Stress of Recovered COVID-19 Patients. Journal of Human Earth and Future, 5(1) 1-18. doi:10.28991/HEF-2024-05-01-01

Copy | Save