fix multi set stat.

dmlc · Apr 8, 2020 · 9b56ca0 · 9b56ca0
1 parent 18f1d99
commit 9b56ca0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/include/xgboost/tree_model.h b/include/xgboost/tree_model.h
@@ -584,7 +584,9 @@ class RegTree : public Model {
     (*this)[nid].SetLeftChild(kInvalidNodeId);
     (*this)[nid].SetRightChild(kInvalidNodeId);
 
-    this->multi_target_stats_.Set(nid, 0, leaf, sum_hess);
+    if (sum_hess.size() != 0) {
+      this->multi_target_stats_.Set(nid, 0, leaf, sum_hess);
+    }
   }
   void SetLeaf(float const& leaf, bst_node_t nid,
                double sum_hess = 0) {