代码之家  ›  专栏  ›  技术社区  ›  HHH

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

  •  1
  • HHH  · 技术社区  · 6 年前

    我在用 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 回复  |  直到 6 年前
        1
  •  1
  •   P.Gupta    5 年前

    DCG和NDCG是比较好的排序算法评价方法。它们可以帮助你衡量结果的质量。你可以在这里看到他们。 https://machinelearningmedium.com/2017/07/24/discounted-cumulative-gain/

    推荐文章