Change obj name to reg:squarederror in learner. (#4427)

* Change memory dump size in R test.
This commit is contained in:
Jiaming Yuan
2019-05-06 21:35:35 +08:00
committed by GitHub
parent 8d1098a983
commit 5de7e12704
2 changed files with 5 additions and 5 deletions

View File

@@ -171,7 +171,7 @@ class LearnerImpl : public Learner {
explicit LearnerImpl(std::vector<std::shared_ptr<DMatrix> > cache)
: cache_(std::move(cache)) {
// boosted tree
name_obj_ = "reg:linear";
name_obj_ = "reg:squarederror";
name_gbm_ = "gbtree";
}
@@ -281,7 +281,7 @@ class LearnerImpl : public Learner {
}
if (cfg_.count("objective") == 0) {
cfg_["objective"] = "reg:linear";
cfg_["objective"] = "reg:squarederror";
}
if (cfg_.count("booster") == 0) {
cfg_["booster"] = "gbtree";