Support XGBoost 1.7

dmlc · Nov 21, 2022 · eb5ad48 · eb5ad48
1 parent 582ad7e
commit eb5ad48
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/frontend/xgboost_json.cc b/src/frontend/xgboost_json.cc
@@ -405,7 +405,8 @@ bool LearnerParamHandler::String(const char *str,
                        output.param.global_bias) ||
           assign_value("num_class", static_cast<unsigned int>(std::max(std::atoi(str), 1)),
                        output.task_param.num_class) ||
-          assign_value("num_feature", std::atoi(str), output.num_feature));
+          assign_value("num_feature", std::atoi(str), output.num_feature) ||
+          check_cur_key("boost_from_average"));
 }
 
 /******************************************************************************