为了提高企业利润,降低运行成本,需要对用户流失行为进行预测,针对可能流失的用户提前进行精准营销,挽留用户。建立XGBoost模型对用户流失数据进行训练,计算出输入特征的重要性排序,选择Top-K特征,得到新训练集。一方面,基于训练集建立贝叶斯优化的XGBoost模型,利用贝叶斯优化寻找最优参数;另一方面,选取8种模型用来建模并验证模型,分别用精确率、准确率、召回率和F1值对模型进行评估。经实验验证,在电信用户流失预测上基于贝叶斯优化的XGBoost模型较其他模型具有更好的预测结果和更高的效率。
In order to improve the profits of enterprises and reduce operating costs, it is necessary to predict the
loss of users, and carry out precise marketing in advance to retain users. The XGBoost model is established to train
the user churn data, and the importance ranking of input features is obtained. The Top-K feature is selected to
obtain a new training set. On the one hand, the XGBoost model of Bayesian optimization is established based on the
training set, and the optimal parameters are found by Bayesian optimization; On the other hand, 8 models are
selected to construct and verify the model, and the model is evaluated in precision, accuracy, recall and F1 value
respectively. Experimental results show that XGBoost model based on Bayesian optimization has better prediction
results and higher efficiency than other models in telecom user churn prediction.