我刚开始使用LightGBM,正在尝试通过LGBMRegressor来拟合一条简单的直线。
import numpy as npimport lightgbm as lgbmxs = 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 抱歉我的英语不好
回答:
设置参数 min_data_in_leaf = 1, min_data_in_bin = 1。