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

「已注销」

于 2018-08-12 16:04:16 发布

阅读量1.9w

点赞数 7

CC 4.0 BY-SA版权

分类专栏：机器学习文章标签： LightGBM GBDT

本文链接：https://blog.youkuaiyun.com/sunyusunyu2011/article/details/81607187

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