使用LightGBM完成训练及预测任务

最新推荐文章于 2025-09-04 11:59:14 发布

原创

最新推荐文章于 2025-09-04 11:59:14 发布 · 1.9w 阅读

83 ·

CC 4.0 BY-SA版权

文章标签：

#LightGBM #GBDT

LightGBM是微软开源的项目，最大的优点是：速度快！！！

之前参加某个竞赛时是使用LightGBM完成的，效果还不错，也很简单，放出来供大家参考~

官方文档：http://lightgbm.apachecn.org/cn/latest/index.html

本文采用的训练集，第一列是Y标签，之后每列是处理好的特征，具体选用哪些特征根据实际情况选择即可。

# -*- coding: utf-8 -*-
# author: Yu Sun

import pandas as pd
import lightgbm as lgb
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split

params = {
    'task': 'train',
    'boosting_type': 'gbdt',  # GBDT算法为基础
    'objective': 'binary',  # 因为要完成预测用户是否买单行为，所以是binary，不买是0，购买是1
    'metric': 'auc',  # 评判指标
    'max_bin': 255,  # 大会有更准的效果,更慢的速度
    'learning_rate': 0.1,  # 学习率
    'num_leaves': 64,  # 大会更准,但可能过拟合
    'max_depth':