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

Support dask dataframe as y for classifier. (#5077)

Browse Source 
* Support dask dataframe as y for classifier.

* Lint.
This commit is contained in:
Jiaming Yuan
2019-12-02 11:53:30 +08:00
committed by GitHub
parent b9dbfe0931
commit 761e938dbe
2 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/python/test_with_dask.py
View File

@@ -109,3 +109,16 @@ def test_classifier(client):
assert list(history['validation_0'].keys())[0] == 'merror'
assert len(list(history['validation_0'])) == 1
assert len(history['validation_0']['merror']) == 2
assert classifier.n_classes_ == 10
# Test with dataframe.
X_d = dd.from_dask_array(X)
y_d = dd.from_dask_array(y)
classifier.fit(X_d, y_d)
assert classifier.n_classes_ == 10
prediction = classifier.predict(X_d)
assert prediction.ndim == 1
assert prediction.shape[0] == kRows