A hierarchy-based machine learning model for happiness prediction

Zongwen Fan,Fenlin Wu,Yaxuan Tang
DOI: https://doi.org/10.1007/s10489-022-03811-x
IF: 5.3
2022-07-14
Applied Intelligence
Abstract:Happiness is one of the most important forms of well-being. It is closely related to personal growth, health, and development. The study of happiness is crucial to providing a way for people to evaluate their happiness based on certain living conditions (or any other conditions) obtained. Thus, to accurately predict happiness based on certain provided living conditions, we propose a hierarchical machine learning model with a two-layer structure for happiness prediction. In the first layer, the parameters of base estimators are tuned using the grid search technique. In the second layer, we optimize the weights of base estimators using particle swarm optimization (PSO). In addition, unlike traditional threshold values (predicted probability) using 0.5 for binary classification, we optimize this threshold by setting its value to between 0 and 1 to make it more applicable in real-world applications. The experimental results indicate that our proposed model optimized by PSO has the best performance, outperforming the models being compared from 2% to 11% in terms of accuracy or F1-score. In addition, the statistical results based on the Wilcoxon rank-sum test verify that our proposed model is significantly better than the other models we compared it to in terms of F1-score. The results confirm that our two-layer hierarchy structure effectively improves the prediction model by aggregating multiple base estimators for happiness prediction. This indicates that our proposed model could provide some helpful information about happiness prediction for decision-makers.
computer science, artificial intelligence
What problem does this paper attempt to address?