当前位置:   article > 正文

LGBM函数及参数详解_lgbm参数

lgbm参数

LGBM Python API

Dataset

class lightgbm.Dataset(data, label=None, max_bin=None, reference=None, weight=None, group=None, init_score=None, silent=False, feature_name='auto', categorical_feature='auto', params=None, free_raw_data=True)

创建一个带label的训练集和交叉验证集

  1. trn_data = lgb.Dataset(X_tr, label=y_tr)
  2. val_data = lgb.Dataset(X_valid, label=y_valid)

Booster 

class lightgbm.Booster(params=None, train_set=None, model_file=None, silent=False)    #booster可用lgb来代替

  • params   字典形式的参数
  • train_set Training dataset.
  • model_file model文件的路径
  • silent 构建模型时是否打印信息

只要有lgb实例,就可以调用下列函数 ,包括 predict,

lgb.add_valid  添加交叉训练集 

lgb.attr(key)  Get attribute string from the Booster.

lgb.current_iteration()  Get the index of the current iteration.

lgb.dump_model(num_iteration=-1)   Dump Booster to jso

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/喵喵爱编程/article/detail/801573
推荐阅读
相关标签
  

闽ICP备14008679号