数据竞赛——3

“达观杯”文本智能处理挑战赛之三

LightGBM:Light Gradient Boosting Machine,是一个实现GBDT (Gradient Boosting Decision Tree)算法的框架,支持高效率的并行训练,并且具有以下优点:

● 更快的训练速度
● 更低的内存消耗
● 更好的准确率
● 分布式支持,可以快速处理海量数据

LightGBM的教程,参见LightGBM 中文文档。

此处的具体实现

import lightgbm as lgb

y_train = Y_train  
y_test = Y_test   
X_train = x_train  
X_test = x_test 

# create dataset for lightgbm
lgb_train = lgb.Dataset(X_train, y_train)
lgb_eval = lgb.Dataset(X_test, y_test, reference=lgb_train)


params = {
    'boosting_type': 'gbdt',
    'objective': 'multiclass',
    'lambda_l1': 0.001,
    'lambda_l2': 0.5,
    'num_leaves': 20,
	'num_trees': 100,
    'learning_rate': 0.01,
    'feature_fraction': 0.9,
    'bagging_fraction': 0.8,
    'bagging_freq': 5,
    'verbose': 0
}

gbm = lgb.train(params,
                lgb_train,
                num_boost_round=100,
                valid_sets=lgb_train)
                
y_pred = gbm.predict(test)

在这里插入图片描述

参考文献
[1]: https://zhuanlan.zhihu.com/p/33700459
[2]: https://blog.csdn.net/anshuai_aw1/article/details/83275299
[3]: https://www.msra.cn/zh-cn/news/features/lightgbm-20170105
[4]: https://github.com/Microsoft/LightGBM
[5]: https://blog.csdn.net/luanpeng825485697/article/details/80236759
[6]: https://blog.csdn.net/feifeizhongda/article/details/81276954

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值