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

Fix wrong expected feature types (#1646)

Browse Source
This commit is contained in:
Yuan (Terry) Tang 2016-10-09 00:16:29 -04:00 committed by Tianqi Chen
parent 485b6c86cc
commit a64fd74421
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/python/test_with_pandas.py
View File

@ -66,10 +66,9 @@ class TestPandas(unittest.TestCase):
[2., 0., 1., 0.],
[3., 0., 0., 1.]])
np.testing.assert_array_equal(result, exp)
dm = xgb.DMatrix(dummies)
assert dm.feature_names == ['B', 'A_X', 'A_Y', 'A_Z']
assert dm.feature_types == ['int', 'float', 'float', 'float']
assert dm.feature_types == ['int', 'int', 'int', 'int']
assert dm.num_row() == 3
assert dm.num_col() == 4