[backport] Fix default metric configuration. (#9575) (#9590)

This commit is contained in:
Jiaming Yuan
2023-09-18 23:40:43 +08:00
committed by GitHub
parent 096047c547
commit ace7713201
6 changed files with 105 additions and 25 deletions

View File

@@ -268,6 +268,13 @@ class PseudoHuberRegression : public FitIntercept {
}
FromJson(in["pseudo_huber_param"], &param_);
}
[[nodiscard]] Json DefaultMetricConfig() const override {
CHECK(param_.GetInitialised());
Json config{Object{}};
config["name"] = String{this->DefaultEvalMetric()};
config["pseudo_huber_param"] = ToJson(param_);
return config;
}
};
XGBOOST_REGISTER_OBJECTIVE(PseudoHuberRegression, "reg:pseudohubererror")