Send default configuration from metric to objective. (#8760)

This commit is contained in:
Jiaming Yuan
2023-02-09 20:18:07 +08:00
committed by GitHub
parent 5f76edd296
commit 199c421d60
4 changed files with 62 additions and 7 deletions

View File

@@ -134,6 +134,12 @@ class AFTObj : public ObjFunction {
void LoadConfig(Json const& in) override {
FromJson(in["aft_loss_param"], &param_);
}
Json DefaultMetricConfig() const override {
Json config{Object{}};
config["name"] = String{this->DefaultEvalMetric()};
config["aft_loss_param"] = ToJson(param_);
return config;
}
private:
AFTParam param_;