From 9d19e13ed030407756585c93200dea1011519143 Mon Sep 17 00:00:00 2001
From: Oleg Sofrygin <osofr@users.noreply.github.com>
Date: Thu, 16 Mar 2017 10:36:57 -0700
Subject: [PATCH] adding a copy of base_margin to slice, fixes a bug where
 base_margin was notcopied during cross-validation (#2007)

---
 src/c_api/c_api.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/c_api/c_api.cc b/src/c_api/c_api.cc
index f1c02188f..d43f9fe00 100644
--- a/src/c_api/c_api.cc
+++ b/src/c_api/c_api.cc
@@ -413,6 +413,9 @@ XGB_DLL int XGDMatrixSliceDMatrix(DMatrixHandle handle,
     if (src.info.weights.size() != 0) {
       ret.info.weights.push_back(src.info.weights[ridx]);
     }
+    if (src.info.base_margin.size() != 0) {
+      ret.info.base_margin.push_back(src.info.base_margin[ridx]);
+    }
     if (src.info.root_index.size() != 0) {
       ret.info.root_index.push_back(src.info.root_index[ridx]);
     }