Supervised machine learning models for depression sentiment analysis

被引:11
作者
Obagbuwa, Ibidun Christiana [1 ]
Danster, Samantha [1 ]
Chibaya, Onil Colin [1 ]
机构
[1] Sol Plaatje Univ, Sch Nat & Appl Sci, Dept Comp Sci & Informat Technol, Kimberley, South Africa
来源
FRONTIERS IN ARTIFICIAL INTELLIGENCE | 2023年 / 6卷
关键词
Twitter; depression; sentiment analysis; text pre-processing; machine learning techniques; social media; natural language processing; mental health;
D O I
10.3389/frai.2023.1230649
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
IntroductionGlobally, the prevalence of mental health problems, especially depression, is at an all-time high. The objective of this study is to utilize machine learning models and sentiment analysis techniques to predict the level of depression earlier in social media users' posts. MethodsThe datasets used in this research were obtained from Twitter posts. Four machine learning models, namely extreme gradient boost (XGB) Classifier, Random Forest, Logistic Regression, and support vector machine (SVM), were employed for the prediction task. ResultsThe SVM and Logistic Regression models yielded the most accurate results when applied to the provided datasets. However, the Logistic Regression model exhibited a slightly higher level of accuracy compared to SVM. Importantly, the logistic regression model demonstrated the advantage of requiring less execution time. DiscussionThe findings of this study highlight the potential of utilizing machine learning models and sentiment analysis techniques for early detection of depression in social media users. The effectiveness of SVM and Logistic Regression models, with Logistic Regression being more efficient in terms of execution time, suggests their suitability for practical implementation in real-world scenarios.
引用
收藏
页数:12
相关论文
共 32 条
[1]  
Aliman G., 2022, J Comput Innov Eng Appl, V7, P35
[2]  
Aljabri M., 2022, International Journal on Advanced Science, Engineering and Information Technology, V12, P1206, DOI [https://doi.org/10.18517/IJASEIT.12.3.14724, DOI 10.18517/IJASEIT.12.3.14724]
[3]   Machine Learning-Based Approach for Depression Detection in Twitter Using Content and Activity Features [J].
Alsagri, Hatoon S. ;
Ykhlef, Mourad .
IEICE TRANSACTIONS ON INFORMATION AND SYSTEMS, 2020, E103D (08) :1825-1832
[4]   Sentiment Analysis in Social Media Data for Depression Detection Using Artificial Intelligence: A Review [J].
Babu N.V. ;
Kanaga E.G.M. .
SN Computer Science, 2022, 3 (1)
[5]  
Dave G. H., 2023, VIDHYAYANA INT MULTI, V8, P777
[6]  
Draelos R., 2019, Glass Box Machine Learning
[7]  
Gaikwad S., 2014, INT J ADV RES COMPUT, V3, P413
[8]  
Jagadishwari V., 2021, Journal of Physics: Conference Series, DOI 10.1088/1742-6596/2070/1/012079
[9]  
Jain Pratyaksh, 2022, Journal of Physics: Conference Series, V2161, DOI 10.1088/1742-6596/2161/1/012034
[10]  
JavaTpoint, 2011, ABOUT US