摘要:
去年做项目用了下XGBoost,今天看到一篇用LightGBM的论文,似乎这个算法效果也挺好的,于是在GPT的帮助下写了个LightGBM的demo,记录一下。
关于LightGBM的介绍我就不多说了,百度谷歌一搜一大堆。懒得查的话可以问GPT,或者看一下这篇介绍:
上面那篇文章很详细的介绍了LightGBM的原理,超参数等等,也给了代码demo,各位感兴趣的也可以直接参考上文。下面是我自己根据我的思路写的代码。这里以sklearn中的iris数据集为例。