Is Your Marriage Reliable?: Divorce Analysis with Machine Learning Algorithms

作者: Jue Kong , Tianrui Chai

DOI: 10.1145/3404555.3404559

关键词:

摘要: 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.

参考文章(7)
Corinna Cortes, Vladimir Vapnik, Support-Vector Networks Machine Learning. ,vol. 20, pp. 273- 297 ,(1995) , 10.1023/A:1022627411411
Chunkai Zhang, Ying Zhou, Jianwei Guo, Guoquan Wang, Xuan Wang, Research on classification method of high-dimensional class-imbalanced datasets based on SVM International Journal of Machine Learning and Cybernetics. ,vol. 10, pp. 1765- 1778 ,(2019) , 10.1007/S13042-018-0853-2
Arga Putra Panatagama, Guntur Putra Pratama, Dwi Yoga Wibawa, SocioEmpathy: A Social-Sensitivity Application to Reduce Stress and Depression of Divorce or Domestic Violence Victims international conference on information and communication technology. ,(2018) , 10.1109/ICOICT.2018.8528755
SK Lakshmanaprabu, K Shankar, M Ilayaraja, Abdul Wahid Nasir, V Vijayakumar, Naveen Chilamkurti, None, Random forest for big data classification in the internet of things using optimal features International Journal of Machine Learning and Cybernetics. ,vol. 10, pp. 2609- 2618 ,(2019) , 10.1007/S13042-018-00916-Z
Leo Breiman, Random Forests Machine Learning archive. ,vol. 45, pp. 5- 32 ,(2001) , 10.1023/A:1010933404324
Mustafa Kemal Yöntem, Kemal Adem, Tahsin Ilhan, Serhat Kılıçarslan, None, DIVORCE PREDICTION USING CORRELATION BASED FEATURE SELECTION AND ARTIFICIAL NEURAL NETWORKS Nevşehir Hacı Bektaş Veli Üniversitesi SBE Dergisi. ,vol. 9, pp. 259- 273 ,(2019)
Anand Avati, Andrew Y. Ng, Sanjay Basu, Khanh K. Thai, Alejandro Schuler, Tony Duan, Daisy Yi Ding, NGBoost: Natural Gradient Boosting for Probabilistic Prediction arXiv: Learning. ,(2019)