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

Remove duplicated DMatrix. (#6592)

Browse Source
This commit is contained in:
Jiaming Yuan 2021-01-12 09:36:56 +08:00 committed by GitHub
parent c709f2aaaf
commit 03cd087da1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -653,12 +653,6 @@ class XGBModel(XGBModelBase):
save_best=True)] save_best=True)]
""" """
train_dmatrix = DMatrix(data=X, label=y, weight=sample_weight,
base_margin=base_margin,
missing=self.missing,
nthread=self.n_jobs)
train_dmatrix.set_info(feature_weights=feature_weights)
evals_result = {} evals_result = {}
train_dmatrix, evals = self._wrap_evaluation_matrices( train_dmatrix, evals = self._wrap_evaluation_matrices(

8
tests/python/test_ranking.py
View File

@ -112,8 +112,12 @@ class TestRanking:
Cleanup test artifacts from download and unpacking Cleanup test artifacts from download and unpacking
:return: :return:
""" """
os.remove(cls.dpath + "MQ2008.zip") zip_f = cls.dpath + "MQ2008.zip"
shutil.rmtree(cls.dpath + "MQ2008") if os.path.exists(zip_f):
os.remove(zip_f)
directory = cls.dpath + "MQ2008"
if os.path.exists(directory):
shutil.rmtree(directory)
def test_training(self): def test_training(self):
""" """