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

Add support for cudf.Series (#4891)

Browse Source
This commit is contained in:
Vibhu Jawa
2019-09-25 20:52:28 -07:00
committed by Jiaming Yuan
parent 82ee2317e8
commit 2fa8b359e0
3 changed files with 23 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/python-gpu/test_from_columnar.py
View File

@@ -27,8 +27,8 @@ def dmatrix_from_cudf(input_type, missing=np.NAN):
np_label = np.random.randn(kRows).astype(input_type)
pa_label = pd.DataFrame(np_label)
cd: cudf.DataFrame = cudf.from_pandas(pa)
cd_label: cudf.DataFrame = cudf.from_pandas(pa_label)
cd = cudf.from_pandas(pa)
cd_label = cudf.from_pandas(pa_label).iloc[:, 0]
dtrain = xgb.DMatrix(cd, missing=missing, label=cd_label)
assert dtrain.num_col() == kCols