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