Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

merge changes Jun092023

Browse Source
This commit is contained in:
amdsc21
2023-06-09 22:41:33 +02:00
parent 35cde3b1b2 152e2fb072
commit 5f78360949
63 changed files with 1299 additions and 243 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/metric/elementwise_metric.cu
View File

@@ -499,9 +499,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;