代码之家  ›  专栏  ›  技术社区  ›  Daiver

lightgbm。sklearn。LGBM回归器无法拟合简单线

  •  1
  • Daiver  · 技术社区  · 7 年前

    我对LightGBM很陌生,我正试图通过LGBMRetressor来拟合简单的线。

    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.])
    

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

    对不起,我的英语很差

    1 回复  |  直到 7 年前
        1
  •  2
  •   Viktoriya Malyasova    7 年前

    设置参数min\u data\u in_leaf=1,min\u data\u in_bin=1。