LightGBMLight Gradient Boosting Machine)是一种梯度提升框架,它使用决策树作为基学习器。LightGBM 可以说是在 XGBoost 上做的优化。LightGBM 为高效并行计算而生,它的 Light 体现在以下几个点上:
更快的训练速度
更低的内存使用
支持单机多线程,多机并行计算,以及 GPU 训练
能够处理大规模数据
from lightgbm import LGBMRegressor
具体参数请参考:
LightGBM官网
LightGBM中文文档
一、参数
class lightgbm.