Function build_dataset

tests/python-gpu/test_gpu_pickling.py:20–25

()

def build_dataset() -> Tuple[np.ndarray, np.ndarray]:
    N = 10
    x = np.linspace(0, N * N, N * N)
    x = x.reshape((N, N))
    y = np.linspace(0, N, N)
    return x, y


def save_pickle(bst: Union[xgb.Booster, xgb.XGBModel], path: str) -> None: