文章基本信息

标题：How Machine Learning Classification Accuracy Changes in a Happiness Dataset with Different Demographic Groups
本地全文：下载
作者：Colm Sweeney ; Edel Ennis ; Maurice Mulvenna 等
期刊名称：Computers
电子版ISSN：2073-431X
出版年度：2022
卷号：11
期号：5
页码：83
DOI：10.3390/computers11050083
语种：English
出版社：MDPI Publishing
摘要：This study aims to explore how machine learning classification accuracy changes with different demographic groups. The HappyDB is a dataset that contains over 100,000 happy statements, incorporating demographic information that includes marital status, gender, age, and parenthood status. Using the happiness category field, we test different types of machine learning classifiers to predict what category of happiness the statements belong to, for example, whether they indicate happiness relating to achievement or affection. The tests were initially conducted with three distinct classifiers and the best performing model was the convolutional neural network (CNN) model, which is a deep learning algorithm, achieving an F1 score of 0.897 when used with the complete dataset. This model was then used as the main classifier to further analyze the results and to establish any variety in performance when tested on different demographic groups. We analyzed the results to see if classification accuracy was improved for different demographic groups, and found that the accuracy of prediction within this dataset declined with age, with the exception of the single parent subgroup. The results also showed improved performance for the married and parent subgroups, and lower performances for the non-parent and un-married subgroups, even when investigating a balanced sample.