我正在使用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)
这里合适什么?我没有任何小组,我应该指定些什么吗?
答案 0 :(得分:0)
DCG和NDCG是用于排名算法的良好评估方法。他们可以帮助您衡量结果的质量。您可以在这里阅读有关它们的信息。 https://machinelearningmedium.com/2017/07/24/discounted-cumulative-gain/