Peng, Ke and Peng, Yan (2022) Research on Telecom Customer Churn Prediction Based on GA-XGBoost and SHAP. Journal of Computer and Communications, 10 (11). pp. 107-120. ISSN 2327-5219
jcc_2022112515515919.pdf - Published Version
Download (2MB)
Abstract
To address the prominent problems faced by customer churn in telecom enterprise management, a telecom customer churn prediction model integrating GA-XGBoost and SHAP is proposed. By using the ADASYN algorithm for data processing on the unbalanced sample set; based on the GA-XGBoost model, the XGBoost algorithm is used to construct the telecom customer churn prediction model, and the hyperparameters of the model are optimized by using the genetic algorithm. The experimental results show that compared with traditional machine learning methods such as GBDT, decision tree, KNN and single XGBoost model, the improved XGBoost model has better performance in recall, F1 value and AUC value; the GA-XGBoost model is integrated with SHAP framework to analyze and explain the important features affecting telecom customer churn, which is more in line with the telecom industry to predict customer the actual situation of churn.
Item Type: | Article |
---|---|
Subjects: | STM Library > Medical Science |
Depositing User: | Managing Editor |
Date Deposited: | 15 Apr 2023 07:21 |
Last Modified: | 25 Jan 2024 04:02 |
URI: | http://open.journal4submit.com/id/eprint/1797 |