现代计算机2024,Vol.30Issue(23) :91-96.DOI:10.3969/j.issn.1007-1423.2024.23.018

基于SMOTE与GBDT算法在银行客户流失预测中的应用

Application of SMOTE and GBDT algorithm in bank customer churn prediction

许超 许莉 游凤芹
现代计算机2024,Vol.30Issue(23) :91-96.DOI:10.3969/j.issn.1007-1423.2024.23.018

基于SMOTE与GBDT算法在银行客户流失预测中的应用

Application of SMOTE and GBDT algorithm in bank customer churn prediction

许超 1许莉 2游凤芹1
扫码查看

作者信息

  • 1. 南京理工大学紫金学院计算机与人工智能学院,南京 210000
  • 2. 南京理工大学数学与统计学院,南京 210000
  • 折叠

摘要

为了提前发现银行潜在的流失客户,构建基于SMOTE与GBDT算法客户流失率预测模型.该模型首先进行数据预处理,然后通过分析数据集中各个特征之间的关系,构建对流失率可能产生影响的新特征.接着采用SMOTE算法处理数据分布不平衡问题.最后将数据集输入集成学习GBDT分类器来进行预测.实验结果表明,在处理不平衡数据集分类问题时,与没有进行SMOTE处理的算法相比,基于SMOTE与GBDT算法在AUC、G-mean和F1值指标上都获得了提升.

Abstract

In order to discover potential churn customers of banks in advance,a customer churn rate prediction model based on SMOTE and GBDT algorithm is constructed.The model first performs data preprocessing,and then constructs new features that may affect the churn rate by analyzing the relationship between various features in the data set.Next,the SMOTE algorithm is used to deal with the imbalance of data distribution.Finally,the data set is input into the integrated learning GBDT classifier for prediction.The SMOTE and GBDT algorithms proposed in this article have achieved improvements in AUC,G-mean,and F1 score indicators.

关键词

流失预测/SMOTE算法/GBDT算法/集成学习

Key words

churn prediction/SMOTE algorithm/GBDT algorithm/ensemble learning

引用本文复制引用

出版年

2024
现代计算机
中大控股

现代计算机

影响因子:0.292
ISSN:1007-1423
段落导航相关论文