Implement Python data handler. (#5689)

* Define data handlers for DMatrix. * Throw ValueError in scikit learn interface.
dmlc · May 22, 2020 · 5af8161 · 5af8161
1 parent 646def5
commit 5af8161
Show file tree

Hide file tree

Showing 7 changed files with 746 additions and 405 deletions.
diff --git a/demo/guide-python/sklearn_examples.py b/demo/guide-python/sklearn_examples.py
@@ -14,7 +14,7 @@
 rng = np.random.RandomState(31337)
 
 print("Zeros and Ones from the Digits dataset: binary classification")
-digits = load_digits(2)
+digits = load_digits(n_class=2)
 y = digits['target']
 X = digits['data']
 kf = KFold(n_splits=2, shuffle=True, random_state=rng)

diff --git a/python-package/xgboost/compat.py b/python-package/xgboost/compat.py
@@ -107,7 +107,6 @@ def lazy_isinstance(instance, module, name):
 try:
     from cudf import DataFrame as CUDF_DataFrame
     from cudf import Series as CUDF_Series
-    from cudf import MultiIndex as CUDF_MultiIndex
     from cudf import concat as CUDF_concat
     CUDF_INSTALLED = True
 except ImportError: