Research on Telecom Customer Churn Prediction Based on GA-XGBoost and SHAP

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

[thumbnail of jcc_2022112515515919.pdf] Text
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

Actions (login required)

View Item
View Item