算法库:lightgbm

LightGBM(Light Gradient Boosting Machine) 是微软开源的 梯度提升决策树(GBDT) 框架,主要用于:

  • 分类:预测用户是否流失、是否购买
  • 回归:预测价格、销量
  • 排序:搜索结果排序、推荐系统
  • 特征重要性分析

核心算法

1
2
3
4
5
6
7
8
9
10
11
12
13
数据

决策树1 → 预测

根据误差继续建树

决策树2 → 修正误差

决策树3 → 继续修正

……

最终预测结果

安装

1
pip install lightgbm==4.3.0
部分 作用
pip Python 包管理工具
install 安装软件包
lightgbm 要安装的机器学习库
==4.3.0 固定安装 4.3.0 版本