reduce memory usage.

dmlc · Mar 16, 2023 · 62b6f1f · 62b6f1f
1 parent d15f363
commit 62b6f1f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python-package/xgboost/sklearn.py b/python-package/xgboost/sklearn.py
@@ -1824,8 +1824,12 @@ def _get_qid(
                 "`fit` method. Please remove one of them."
             )
         q_x = X.qid
-        X = X.drop("qid", axis=1)
-        return X, q_x
+        d = {}
+        columns = X.columns.difference(["qid"])
+        for c in columns:
+            d[c] = X.loc[:, c]
+        X_n = type(X)(d)
+        return X_n, q_x
     return X, qid