Implement __sklearn_is_fitted__. (#7230)

2021-09-15 19:09:04 +08:00
parent d997c967d5
commit 037dd0820d
2 changed files with 5 additions and 2 deletions
--- a/tests/python-gpu/test_gpu_with_sklearn.py
+++ b/tests/python-gpu/test_gpu_with_sklearn.py
@@ -19,7 +19,7 @@ def test_gpu_binary_classification():
    from sklearn.datasets import load_digits
    from sklearn.model_selection import KFold

-    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)