Fix default metric configuration. (#9575)

This commit is contained in:
Jiaming Yuan
2023-09-14 04:05:47 +08:00
committed by GitHub
parent b438d684d2
commit 300f9ace06
6 changed files with 105 additions and 25 deletions

View File

@@ -287,6 +287,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")