Python run_boost_from_predictionの例

コード例 #1

ファイル: test_gpu_with_dask.py プロジェクト: r-yuzhe/xgboost

def test_boost_from_prediction(local_cuda_cluster: LocalCUDACluster) -> None:
    from sklearn.datasets import load_breast_cancer
    with Client(local_cuda_cluster) as client:
        X_, y_ = load_breast_cancer(return_X_y=True)
        X = dd.from_array(X_, chunksize=100)
        y = dd.from_array(y_, chunksize=100)
        run_boost_from_prediction(X, y, "gpu_hist", client)

コード例 #2

def test_boost_from_prediction(local_cuda_cluster: LocalCUDACluster) -> None:
    import cudf
    from sklearn.datasets import load_breast_cancer, load_digits
    with Client(local_cuda_cluster) as client:
        X_, y_ = load_breast_cancer(return_X_y=True)
        X = dd.from_array(X_, chunksize=100).map_partitions(cudf.from_pandas)
        y = dd.from_array(y_, chunksize=100).map_partitions(cudf.from_pandas)
        run_boost_from_prediction(X, y, "gpu_hist", client)

        X_, y_ = load_digits(return_X_y=True)
        X = dd.from_array(X_, chunksize=100).map_partitions(cudf.from_pandas)
        y = dd.from_array(y_, chunksize=100).map_partitions(cudf.from_pandas)
        run_boost_from_prediction_multi_clasas(X, y, "gpu_hist", client)