Time and performance comparison on suicide detection using various feature engineering and machine learning models

Abstract

Today more people use social media to express their opinion and their emotions. There are many types of text in social media including text that convey a tendency to be depressed or suicidal. We use sentiment analysis to detect suicidal texts, because if detected, it could save many lives and many families. In this research, we have an objective to explore a method that is both high performance and less time-using. We design experiments that have 30 combinations between five machine learning models with six feature engineering methods. All experiments use accuracy and total time for model generation as metrics. We use deep neural networks with glove embedding as a comparator because this combination performed well in this dataset on Kaggle competition. From the experimental results, we find that the suitable combination that generates fast and has good accuracy is Random Forest with TF-IDF with 0.897 and 145 seconds.