Fix metainfo from DataFrame. (#5216)

* Fix metainfo from DataFrame. * Unify helper functions for data and meta.
2020-01-22 16:29:44 +08:00
parent 5d4c24a1fc
commit 1891cc766d
5 changed files with 225 additions and 237 deletions
--- a/tests/python/test_with_sklearn.py
+++ b/tests/python/test_with_sklearn.py
@@ -97,6 +97,7 @@ def test_ranking():
    valid_data = xgb.DMatrix(x_valid, y_valid)
    test_data = xgb.DMatrix(x_test)
    train_data.set_group(train_group)
+    assert train_data.get_label().shape[0] == x_train.shape[0]
    valid_data.set_group(valid_group)

    params_orig = {'tree_method': 'exact', 'objective': 'rank:pairwise',