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

Rename misspelled config parameter for pseudo-Huber (#7904)

Browse Source
This commit is contained in:
Philip Hyunsu Cho 2022-05-15 06:38:33 -07:00 committed by GitHub
parent 1baad8650c
commit 4cd14aee5a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
doc/model.schema
View File

@ -207,7 +207,7 @@
} }
} }
}, },
"pseduo_huber_param": { "pseudo_huber_param": {
"type": "object", "type": "object",
"properties": { "properties": {
"huber_slope": { "huber_slope": {

4
src/metric/elementwise_metric.cu
View File

@ -171,11 +171,11 @@ class PseudoErrorLoss : public Metric {
public: public:
const char* Name() const override { return "mphe"; } const char* Name() const override { return "mphe"; }
void Configure(Args const& args) override { param_.UpdateAllowUnknown(args); } void Configure(Args const& args) override { param_.UpdateAllowUnknown(args); }
void LoadConfig(Json const& in) override { FromJson(in["pseduo_huber_param"], &param_); } void LoadConfig(Json const& in) override { FromJson(in["pseudo_huber_param"], &param_); }
void SaveConfig(Json* p_out) const override { void SaveConfig(Json* p_out) const override {
auto& out = *p_out; auto& out = *p_out;
out["name"] = String(this->Name()); out["name"] = String(this->Name());
out["pseduo_huber_param"] = ToJson(param_); out["pseudo_huber_param"] = ToJson(param_);
} }
double Eval(const HostDeviceVector<bst_float>& preds, const MetaInfo& info, double Eval(const HostDeviceVector<bst_float>& preds, const MetaInfo& info,

6
src/objective/regression_obj.cu
View File

@ -255,16 +255,16 @@ class PseudoHuberRegression : public ObjFunction {
void SaveConfig(Json* p_out) const override { void SaveConfig(Json* p_out) const override {
auto& out = *p_out; auto& out = *p_out;
out["name"] = String("reg:pseudohubererror"); out["name"] = String("reg:pseudohubererror");
out["pseduo_huber_param"] = ToJson(param_); out["pseudo_huber_param"] = ToJson(param_);
} }
void LoadConfig(Json const& in) override { void LoadConfig(Json const& in) override {
auto const& config = get<Object const>(in); auto const& config = get<Object const>(in);
if (config.find("pseduo_huber_param") == config.cend()) { if (config.find("pseudo_huber_param") == config.cend()) {
// The parameter is added in 1.6. // The parameter is added in 1.6.
return; return;
} }
FromJson(in["pseduo_huber_param"], &param_); FromJson(in["pseudo_huber_param"], &param_);
} }
}; };