Fix weighted samples in multi-class AUC. (#7300)

This commit is contained in:
Jiaming Yuan
2021-10-11 15:12:29 +08:00
committed by GitHub
parent 69d3b1b8b4
commit 298af6f409
6 changed files with 41 additions and 17 deletions

View File

@@ -86,7 +86,7 @@ void CheckRankingObjFunction(std::unique_ptr<xgboost::ObjFunction> const& obj,
xgboost::bst_float GetMetricEval(
xgboost::Metric * metric,
xgboost::HostDeviceVector<xgboost::bst_float> preds,
xgboost::HostDeviceVector<xgboost::bst_float> const& preds,
std::vector<xgboost::bst_float> labels,
std::vector<xgboost::bst_float> weights = std::vector<xgboost::bst_float>(),
std::vector<xgboost::bst_uint> groups = std::vector<xgboost::bst_uint>());