Is Your Marriage Reliable?
Jue Kong, Tianrui Chai · 2020
In recent years, global divorce rate is still high. What kind of couple will divorce and what factors lead to divorce are important problems that worth studying. In this paper, we apply three machine learning algorithms (Support Vector Machine (SVM), Random forest (RF) and Natural Gradient Boosting (NGBoost)) on a divorce prediction dataset. The dataset consists of 170 samples, each of which contains 54 questions about the couple's emotional status. We regard the scores of 54 questions as the features of each sample to apply our machine learning algorithms. Compared with SVM and RF, NGBoost has superior performance as NGBoost can achieve 0.9833 accuracy, 0.9769 precision and 0.9828 F1 score. In addition, we also show the most important features in the model of RF and NGBoost to find the most important factors which lead to divorce.