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

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