/pyLightGBM

Python binding for Microsoft LightGBM

Primary LanguagePython

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等