Update gbdt.cpp

microsoft · Aug 1, 2017 · 399710a · 399710a
1 parent f5aa1c9
commit 399710a
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/boosting/gbdt.cpp b/src/boosting/gbdt.cpp
@@ -48,6 +48,7 @@ GBDT::GBDT()
     num_threads_ = omp_get_num_threads();
   }
   average_output_ = false;
+  tree_learner_ = nullptr;
 }
 
 GBDT::~GBDT() {
@@ -217,13 +218,12 @@ void GBDT::ResetTrainingData(const Dataset* train_data, const ObjectiveFunction*
 
 void GBDT::ResetConfig(const BoostingConfig* config) {
   auto new_config = std::unique_ptr<BoostingConfig>(new BoostingConfig(*config));
-
   early_stopping_round_ = new_config->early_stopping_round;
   shrinkage_rate_ = new_config->learning_rate;
-
-  ResetBaggingConfig(new_config.get());
-
-  tree_learner_->ResetConfig(&new_config->tree_config);
+  if (tree_learner_ != nullptr) {
+    ResetBaggingConfig(new_config.get());
+    tree_learner_->ResetConfig(&new_config->tree_config);
+  }
   gbdt_config_.reset(new_config.release());
 }