Implement Python data handler. (#5689)

* Define data handlers for DMatrix.
* Throw ValueError in scikit learn interface.
This commit is contained in:
Jiaming Yuan
2020-05-22 11:53:55 +08:00
committed by GitHub
parent 646def51e0
commit 5af8161a1a
7 changed files with 746 additions and 405 deletions

View File

@@ -14,7 +14,7 @@ from sklearn.datasets import load_iris, load_digits, load_boston
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)