- The new implementation is more strict as only binary labels are accepted. The previous implementation converts values greater than 1 to 1. - Deterministic GPU. (no atomic add). - Fix top-k handling. - Precise definition of MAP. (There are other variants on how to handle top-k). - Refactor GPU ranking tests.
12 lines
243 B
C++
12 lines
243 B
C++
/**
|
|
* Copyright 2023 by XGBoost Contributors
|
|
*/
|
|
#pragma once
|
|
#include <xgboost/context.h> // for Context
|
|
|
|
namespace xgboost::ltr {
|
|
void TestNDCGCache(Context const* ctx);
|
|
|
|
void TestMAPCache(Context const* ctx);
|
|
} // namespace xgboost::ltr
|