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

Fix multi-class loading

Browse Source
This commit is contained in:
tqchen
2016-03-10 19:21:29 -08:00
parent 96b17971ac
commit ec2fb5bc48
3 changed files with 31 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/learner.cc
View File

@@ -125,8 +125,8 @@ class LearnerImpl : public Learner {
}
void Configure(const std::vector<std::pair<std::string, std::string> >& args) override {
tparam.InitAllowUnknown(args);
// add to configurations
tparam.InitAllowUnknown(args);
cfg_.clear();
for (const auto& kv : args) {
if (kv.first == "eval_metric") {
@@ -187,6 +187,8 @@ class LearnerImpl : public Learner {
// set number of features correctly.
cfg_["num_feature"] = common::ToString(mparam.num_feature);
cfg_["num_class"] = common::ToString(mparam.num_class);
if (gbm_.get() != nullptr) {
gbm_->Configure(cfg_.begin(), cfg_.end());
}