lightgbm - LightGBM ranker 函数使用什么评估指标

Question

我正在使用LGMRankerfromLightGBM但不确定我应该使用什么评估指标。这是我的代码：

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)

这里有什么合适的？我没有任何组，我应该指定一些东西吗？

标签： lightgbmboosting

Answer 1

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