UIJRT » United International Journal for Research & Technology

Opinion Analysis and Machine Learning Modeling for Depression Detection

PAPER DETAILS

CITE THIS

Baale, A.A., Olasunkanmi, O.R., Adelodun, F.E. and Adigun, A.A., 2021. Opinion Analysis and Machine Learning Modeling for Depression Detection. United International Journal for Research & Technology (UIJRT), 2(4), pp.38-43.

Abstract

Many people express opinions on social media sites when they suffer from mental disorders like depression, anxiety, and tension due to pressures, external environment, and other reasons. Such posts shared via Twitter, Facebook, and Instagram are used to identify a person’s state of mind. The situation ideation, which is rarely noticed on time until after a tragic consequence, are often earlier expressed overtly or covertly in social media posts. As a result, this research aimed at implementing four (4) classifiers- Logistic Regression (LR), Naive Bayes (NB), Random Forest (RF), and Decision Tree (DT) on two text-feature extraction techniques- Term Frequency- Inverse Document Frequency (TF-IDF) and Bag of Words (BOW). We split the Sentiment140 downloaded dataset from Kaggle into 75%, 25% training, and testing data to predict mental health depression in the tweet’s dataset. TF-IDF models produced the highest accuracy with DT (99%) and RF (99%), while the BOW extends the same performance with LR (99%). However, to mitigate the challenges of erroneous classification of depressive individuals as neutral, Receiver Operating Characteristic / Area Under Curve (ROC_AUC) scores of classifiers used was obtained. At the same time, the RF and DT produced 99%, the highest ROC_AUC score. Overall performance of models revealed that tree-based models performed better on the test data used in this research to classify and predict mental health depression in the tweet’s dataset.

Keywords: Bag of Words, Depression, Social Media, Machine Learning.

Related Papers

For Conference & Paper Publication​