9、LightGBM、XGBoost与深度学习对比及Optuna参数优化

LightGBM、XGBoost与深度学习对比及Optuna参数优化

1. LightGBM、XGBoost与TabTransformer对比

1.1 人口收入数据集实验

对LightGBM、XGBoost和TabTransformer三种算法都使用网格搜索技术进行参数优化。对于两种提升算法(LightGBM和XGBoost),优化了学习率、分箱大小和树的数量;对于TabTransformer,既要优化参数(如学习率、权重衰减和丢弃率),也要优化架构方面(如变压器块的数量和多层感知机(MLP)中的隐藏层数量)。

以下是在人口收入数据集上训练三种模型的验证集结果:
| 模型 | 训练时间 | 准确率 | F1分数 |
| — | — | — | — |
| LightGBM GBDT | 1.05s | 84.46% | 0.71 |
| XGBoost GBDT | 5.5s | 84.44% | 0.72 |
| TabTransformer | 113.63s | 77.00% | 0.64 |

可以看出,XGBoost和LightGBM在该数据集上表现相似,准确率达到84%,F1分数约为0.7。TabTransformer模型表现较差,准确率和F1分数较低。在训练时间方面,LightGBM比其他方法快得多,比XGBoost快5.23倍,比TabTransformer快108.22倍。

1.2 信用卡欺诈检测实验

1.2.1 数据集介绍

第二个任务是在信用卡交易数据集中检测欺诈交易。数据集可在https://www.kaggle.com

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符  | 博主筛选后可见
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值