91db788cc3fa60962a159e9002a20b8930b6e314,dask_xgboost/tests/test_core.py,,test_dask,#Any#Any#Any#Any#,16

Before Change

@gen_cluster(client=True, Worker=Nanny)
def test_dask(c, s, a, b):
    x = np.random.random((1000, 10))
    y = np.random.random(1000)

    dtrain = xgb.DMatrix(x, label=y)

After Change

    dtrain = xgb.DMatrix(df, label=labels)
    bst = xgb.train(param, dtrain)

    ddf = dd.from_pandas(df, npartitions=4)
    dlabels = dd.from_pandas(labels, npartitions=4)
    dbst = yield dxgb._train(c, param, ddf, dlabels)

    result = bst.predict(dtrain)
    dresult = dbst.predict(dtrain)

    correct = (result > 0.5) == labels
    dcorrect = (dresult > 0.5) == labels
    assert dcorrect.sum() >= correct.sum()


def test_synchronous_api(loop):

Instances