选项+LightGBM\=EightGBM
OptGBM的Python项目详细描述
OptGBM公司
OptGBM(=Optuna+LightGBM)提供了一个scikit学习兼容估计器,它使用Optuna调整LightGBM中的超参数。在
示例
importoptgbmaslgbfromsklearn.datasetsimportload_bostonreg=lgb.LGBMRegressor(random_state=0)X,y=load_boston(return_X_y=True)reg.fit(X,y)y_pred=reg.predict(X,y)
默认情况下,将搜索以下超参数。在
bagging_fraction
bagging_freq
feature_fractrion
lambda_l1
lambda_l2
max_depth
min_data_in_leaf
num_leaves
安装
^{pr2}$测试
tox
- 项目
标签: