LightGBM等级函数使用什么评估指标

时间:2019-01-17 22:00:08

标签: lightgbm boosting

我正在使用LGMRanker中的LightGBM,但不确定应该使用哪种评估指标。这是我的代码:

import lightgbm as lgb


gbm = lgb.LGBMRanker
gridParams = {
'learning_rate': [0.005,0.01,0.02],
'max_depth': [5,6,7],
'n_estimators': [100,200],
'num_leaves': [20,30,50]
}
lgb_grid = GridSearchCV(estimator = gbm, param_grid = gridParams, scoring = '??', cv = 3, verbose=2, n_jobs = -1)

这里合适什么?我没有任何小组,我应该指定些什么吗?

1 个答案:

答案 0 :(得分:0)

DCG和NDCG是用于排名算法的良好评估方法。他们可以帮助您衡量结果的质量。您可以在这里阅读有关它们的信息。 https://machinelearningmedium.com/2017/07/24/discounted-cumulative-gain/