首页 | 本学科首页   官方微博 | 高级检索  
     检索      

基于代价敏感LightGBM的网购意愿预测研究
引用本文:罗咪,邱一卉,林建宗.基于代价敏感LightGBM的网购意愿预测研究[J].厦门大学学报(自然科学版),2024(2):232-240.
作者姓名:罗咪  邱一卉  林建宗
作者单位:厦门理工学院经济与管理学院
基金项目:国家自然科学基金(7180040248);;福建省自然科学基金(2022J011261);
摘    要:目的]由于购买商品的消费者数量远小于未购买商品的消费者数量,网购意愿预测研究是典型的不平衡数据分类问题.研究不平衡数据的分类问题以提升网购意愿预测的分类准确率,该问题主要存在少数类样本识别准确率远小于多数类样本的问题.方法]提出一种基于贝叶斯优化的代价敏感轻量梯度提升机(Light Gradient Boosting Machine, LightGBM)模型.首先引入误分类代价作为惩罚因子修正LightGBM的损失函数,其次通过阈值移动降低模型的分类阈值以提高针对少数类样本的预测准确率,最后利用贝叶斯优化算法优化误分类代价参数、分类阈值及其他参数.结果]从KEEL数据库中选取5个典型的不平衡数据集进行对比实验,相较于标准LightGBM模型,改进LightGBM模型的AUC值和G-mean值均提升了10%左右;相较于遗传算法优化代价敏感LightGBM模型和粒子群优化代价敏感LightGBM模型,改进LightGBM模型的AUC值和G-mean值普遍提升了4%左右;相较于ADASYN-LightGBM模型和BorderlineSMOTE-LightGBM模型,改进LightGBM...

关 键 词:不平衡数据  贝叶斯优化  代价敏感  LightGBM  网购意愿预测
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号