Какой показатель оценки использовать для функции ранжирования LightGBM

Question

Я использую 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)

Что здесь уместно? У меня нет какой-либо группы, я должен указать что-то?

Answer 1

DCG и NDCG - хорошие методы оценки для алгоритмов ранжирования. Они могут помочь вам измерить качество ваших результатов. Вы можете прочитать о них здесь. https://machinelearningmedium.com/2017/07/24/discounted-cumulative-gain/