1

我对 LightGBM 很陌生，我正在尝试通过 LGBMRegressor 拟合简单的线条。

import numpy as np
import lightgbm as lgbm
xs = np.linspace(0, 10, 30).reshape((-1, 1)) 
ys = np.linspace(0, 10, 30)    
reg = lgbm.sklearn.LGBMRegressor()
reg.fit(xs, ys) 
print reg.predict(xs)
>>> array([ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,
    0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,  0.,
    0.,  0.,  0.,  0.])

我做错了什么？也许我需要更多功能？

我的 LightGBM 版本是 2.0.7（来自 github 源），python 版本是 2.7.12

PS对不起我的英语不好

4

1 回答 1

2

设置参数 min_data_in_leaf = 1, min_data_in_bin = 1。

于 2017-12-16T18:40:45.230 回答