Journal of Human, Earth, and Future (Mar 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

DOI
https://doi.org/10.28991/HEF-2024-05-01-01
Journal volume & issue
Vol. 5, no. 1
pp. 1 – 18

Abstract

Read online

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. Doi: 10.28991/HEF-2024-05-01-01 Full Text: PDF

Keywords