wepe / pyLightGBM

Python binding for Microsoft LightGBM

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

LightGBM 的 issue上在讨论python binding,看到了这个quick wrapper pyLightGBM,实现挺简单的

  • models.py里构造了GenericGMB类,继承了sklearn的BaseEstimator
  • 构造函数里传入各种参数,然后写入 train.config
  • fit函数里用python的 os.system("./lightgbm config=train.conf")去调用cpp 编译出来的可执行文件lightgbm

因为继承了sklearn的BaseEstimator,可以使用一些sklearn的特性,像cv,cross_val_score等

About

Python binding for Microsoft LightGBM


Languages

Language:Python 53.6%Language:Jupyter Notebook 46.4%