From 49e641012fc11c6781140113a7a49364de8c6bc8 Mon Sep 17 00:00:00 2001
From: tqchen <tianqi.tchen@gmail.com>
Date: Mon, 30 Mar 2015 16:05:51 -0700
Subject: [PATCH] add objective

---
 src/learner/learner-inl.hpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/learner/learner-inl.hpp b/src/learner/learner-inl.hpp
index 0d2331de1..c4b20e4de 100644
--- a/src/learner/learner-inl.hpp
+++ b/src/learner/learner-inl.hpp
@@ -170,6 +170,9 @@ class BoostLearner : public rabit::ISerializable {
     if (gbm_ != NULL) delete gbm_;
     this->InitTrainer(calc_num_feature);
     this->InitObjGBM();
+    char tmp[32];
+    utils::SPrintf(tmp, sizeof(tmp), "%u", mparam.num_class);
+    obj_->SetParam("num_class", tmp);
     gbm_->LoadModel(fi, with_pbuffer);
     if (!with_pbuffer || distributed_mode == 2) {
       gbm_->ResetPredBuffer(pred_buffer_size);