fix bug in reset config for dataset (#2453)

* Update feature_histogram.hpp * Update feature_histogram.hpp
microsoft · Sep 26, 2019 · 7a8c4e5 · 7a8c4e5
1 parent 70fc45b
commit 7a8c4e5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/treelearner/feature_histogram.hpp b/src/treelearner/feature_histogram.hpp
@@ -748,13 +748,16 @@ class HistogramPool {
       OMP_LOOP_EX_END();
     }
     OMP_THROW_EX();
+    train_data_ = train_data;
   }
 
   void ResetConfig(const Config* config) {
     int size = static_cast<int>(feature_metas_.size());
     #pragma omp parallel for schedule(static, 512) if (size >= 1024)
     for (int i = 0; i < size; ++i) {
       feature_metas_[i].config = config;
+      feature_metas_[i].monotone_type = train_data_->FeatureMonotone(i);
+      feature_metas_[i].penalty = train_data_->FeaturePenalte(i);
     }
   }
   /*!
@@ -823,6 +826,7 @@ class HistogramPool {
   std::vector<int> inverse_mapper_;
   std::vector<int> last_used_time_;
   int cur_time_ = 0;
+  const Dataset* train_data_;
 };
 
 }  // namespace LightGBM