Move to c++.

dmlc · Dec 15, 2020 · 29e6de4 · 29e6de4
1 parent a59daaa
commit 29e6de4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/python-package/xgboost/core.py b/python-package/xgboost/core.py
@@ -786,8 +786,6 @@ def slice(
                 ctypes.c_int(1 if allow_groups else 0),
             )
         )
-        res.feature_names = self.feature_names
-        res.feature_types = self.feature_types
         return res
 
     @property

diff --git a/src/data/data.cc b/src/data/data.cc
@@ -299,6 +299,11 @@ MetaInfo MetaInfo::Slice(common::Span<int32_t const> ridxs) const {
 
   out.feature_weigths.Resize(this->feature_weigths.Size());
   out.feature_weigths.Copy(this->feature_weigths);
+
+  out.feature_names = this->feature_names;
+  out.feature_types.Copy(this->feature_types);
+  out.feature_type_names = this->feature_type_names;
+
   return out;
 }