fix #1089 : voting parallel bug for sparse data

microsoft · Dec 1, 2017 · 159668d · 159668d
1 parent 63b6082
commit 159668d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/treelearner/voting_parallel_tree_learner.cpp b/src/treelearner/voting_parallel_tree_learner.cpp
@@ -169,7 +169,7 @@ void VotingParallelTreeLearner<TREELEARNER_T>::GlobalVoting(int leaf_idx, const
   }
   // get mean number on machines
   score_t mean_num_data = GetGlobalDataCountInLeaf(leaf_idx) / static_cast<score_t>(num_machines_);
-  std::vector<LightSplitInfo> feature_best_split(this->num_features_, LightSplitInfo());
+  std::vector<LightSplitInfo> feature_best_split(this->train_data_->num_total_features() , LightSplitInfo());
   for (auto & split : splits) {
     int fid = split.feature;
     if (fid < 0) {