add default to poisson -> max_delta_step to enable loading/saving/dumping of model (#1781)
This commit is contained in:
parent
58aa1129ea
commit
f11f2bd5fd
@ -150,7 +150,7 @@ XGBOOST_REGISTER_OBJECTIVE(LogisticRaw, "binary:logitraw")
|
|||||||
struct PoissonRegressionParam : public dmlc::Parameter<PoissonRegressionParam> {
|
struct PoissonRegressionParam : public dmlc::Parameter<PoissonRegressionParam> {
|
||||||
float max_delta_step;
|
float max_delta_step;
|
||||||
DMLC_DECLARE_PARAMETER(PoissonRegressionParam) {
|
DMLC_DECLARE_PARAMETER(PoissonRegressionParam) {
|
||||||
DMLC_DECLARE_FIELD(max_delta_step).set_lower_bound(0.0f)
|
DMLC_DECLARE_FIELD(max_delta_step).set_lower_bound(0.0f).set_default(0.7f)
|
||||||
.describe("Maximum delta step we allow each weight estimation to be." \
|
.describe("Maximum delta step we allow each weight estimation to be." \
|
||||||
" This parameter is required for possion regression.");
|
" This parameter is required for possion regression.");
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user