模型托管>> GBDT-LR>> 返回
项目作者： morenjiujiu

项目描述 ：
GBDT结合LR的二分类模型，封装成了一个类。scikit-learn风格，可以fit和predict。有run_demo
高级语言： Python
项目地址: git://github.com/morenjiujiu/GBDT-LR.git
创建时间： 2019-09-04T08:34:06Z
项目社区：https://github.com/morenjiujiu/GBDT-LR
开源协议：
下载


GBDTLR Estimator

  • GBDT结合LR的二分类模型，封装成了一个类GBDTLR。scikit-learn风格，可以fit和predict。
  • GBDTLR这个类中，我只挑选了GBDT的6个参数，LR的2个参数。大家可以根据需要把其他参数补全。

运行环境

  • python3
  • numpy, pandas, sklearn

注意几个坑

  • 要继承自BaseEstimator，否则会有 get_params 这种错误
  • init()中 “self.n_estimators=n_estimators” 的那些self不能省略，如果只有 self.gbdt_params 和 self.lr_params 是不行的

参考

GBDT+LR的背景

创建自定义的estimator，scikit-learn风格