Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove cupy.array_equal, since it's not compatible with cuPy 7.8 (#6528) (#6535)

Browse Source 
Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>
This commit is contained in:
Jiaming Yuan 2020-12-20 15:11:50 +08:00 committed by GitHub
parent d95d02132a
commit 76c361431f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
python-package/xgboost/sklearn.py
View File

@ -841,14 +841,18 @@ class XGBClassifier(XGBModel, XGBClassifierBase):
self.classes_ = cp.unique(y.values) self.classes_ = cp.unique(y.values)
self.n_classes_ = len(self.classes_) self.n_classes_ = len(self.classes_)
can_use_label_encoder = False can_use_label_encoder = False
if not cp.array_equal(self.classes_, cp.arange(self.n_classes_)): expected_classes = cp.arange(self.n_classes_)
if (self.classes_.shape != expected_classes.shape or
not (self.classes_ == expected_classes).all()):
raise ValueError(label_encoding_check_error) raise ValueError(label_encoding_check_error)
elif _is_cupy_array(y): elif _is_cupy_array(y):
import cupy as cp # pylint: disable=E0401 import cupy as cp # pylint: disable=E0401
self.classes_ = cp.unique(y) self.classes_ = cp.unique(y)
self.n_classes_ = len(self.classes_) self.n_classes_ = len(self.classes_)
can_use_label_encoder = False can_use_label_encoder = False
if not cp.array_equal(self.classes_, cp.arange(self.n_classes_)): expected_classes = cp.arange(self.n_classes_)
if (self.classes_.shape != expected_classes.shape or
not (self.classes_ == expected_classes).all()):
raise ValueError(label_encoding_check_error) raise ValueError(label_encoding_check_error)
else: else:
self.classes_ = np.unique(y) self.classes_ = np.unique(y)