Set ndcg to default for LTR. (#8822)

- Add document.
- Add tests.
- Use `ndcg` with `topk` as default.
This commit is contained in:
Jiaming Yuan
2023-06-09 23:31:33 +08:00
committed by GitHub
parent e4dd6051a0
commit 1fcc26a6f8
18 changed files with 842 additions and 19 deletions

View File

@@ -483,9 +483,13 @@ class QuantileError : public MetricNoCache {
const char* Name() const override { return "quantile"; }
void LoadConfig(Json const& in) override {
auto const& name = get<String const>(in["name"]);
CHECK_EQ(name, "quantile");
FromJson(in["quantile_loss_param"], &param_);
auto const& obj = get<Object const>(in);
auto it = obj.find("quantile_loss_param");
if (it != obj.cend()) {
FromJson(it->second, &param_);
auto const& name = get<String const>(in["name"]);
CHECK_EQ(name, "quantile");
}
}
void SaveConfig(Json* p_out) const override {
auto& out = *p_out;