首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >LightGBM:持续培训技能学习界面

LightGBM:持续培训技能学习界面
EN

Stack Overflow用户
提问于 2018-10-20 03:13:51
回答 1查看 648关注 0票数 0

我正在尝试弄清楚如何使用LightGBM Sklearn接口来继续训练分类器。文档清楚地表明,我需要为fit方法提供一个"init_score“输入。这应该是一个类似数组的形状n_samples,所以在行级上。不幸的是,文档没有解释如何从先前拟合的树计算init_score。在我看来,没有明显的方法来计算这个分数。例如,predict方法在默认情况下生成标签,如果我请求raw_scores,我会得到一个形状为n_samples,n_labels的ndarray。我已经看过了LightGBM代码,但是我不知道如何为init_score创建一个合适的值。有人能帮帮忙吗?

EN

回答 1

Stack Overflow用户

发布于 2018-10-20 06:35:57

下面是为我工作的代码。

代码语言:javascript
复制
X_b, X_test_b, y_b, y_test_b = train_test_split(X,y)
d_train = lgb.Dataset(X_b, label=y_b)

params = {}
params['learning_rate'] = 0.03
params['boosting_type'] = 'gbdt'
params['objective'] = 'regression'
<keep going...>

lg_model = lgb.train(params, d_train, 27500)
y_pred = lg_model.predict(X_test_b)
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/52898616

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档