Kyle Nicholson
e6a238c020
Update base margin dask ( #6155 )
...
* Add `base-margin`
* Add `output_margin` to regressor.
Co-authored-by: fis <jm.yuan@outlook.com>
2020-09-26 21:30:52 +08:00
Jiaming Yuan
33d80ffad0
[dask] Support more meta data on functional interface. ( #6132 )
...
* Add base_margin, label_(lower|upper)_bound.
* Test survival training with dask.
2020-09-21 16:56:37 +08:00
Rory Mitchell
47350f6acb
Allow kwargs in dask predict ( #6117 )
2020-09-15 13:04:03 +12:00
Jiaming Yuan
29b7fea572
Optimize cpu sketch allreduce for sparse data. ( #6009 )
...
* Bypass RABIT serialization reducer and use custom allgather based merging.
2020-08-19 10:03:45 +08:00
Jiaming Yuan
ee70a2380b
Unify CPU hist sketching ( #5880 )
2020-08-12 01:33:06 +08:00
jameskrach
bd6b7f4aa7
[Breaking] Fix .predict() method and add .predict_proba() in xgboost.dask.DaskXGBClassifier ( #5986 )
2020-08-11 16:11:28 +08:00
Jiaming Yuan
801e6b6800
Fix dask predict shape infer. ( #5989 )
2020-08-08 14:29:22 +08:00
Jiaming Yuan
fa3715f584
[Dask] Asyncio support. ( #5862 )
2020-07-30 06:23:58 +08:00
Jiaming Yuan
7c2686146e
Dask device dmatrix ( #5901 )
...
* Fix softprob with empty dmatrix.
2020-07-17 13:17:43 +08:00
Jiaming Yuan
6601a641d7
Thread safe, inplace prediction. ( #5389 )
...
Normal prediction with DMatrix is now thread safe with locks. Added inplace prediction is lock free thread safe.
When data is on device (cupy, cudf), the returned data is also on device.
* Implementation for numpy, csr, cudf and cupy.
* Implementation for dask.
* Remove sync in simple dmatrix.
2020-03-30 15:35:28 +08:00
Jiaming Yuan
cd7d6f7d59
[dask] Fix missing value for scikit-learn interface. ( #5435 )
2020-03-20 10:56:01 -04:00
Jiaming Yuan
760d5d0c3c
[dask] Accept other inputs for prediction. ( #5428 )
...
* Returns a series when input is dataframe.
* Merge assert client.
2020-03-19 17:05:55 +08:00
Jiaming Yuan
b51124c158
[dask] Enable gridsearching with skl. ( #5417 )
2020-03-16 04:51:51 +08:00
Jiaming Yuan
761a5dbdfc
[dask] Honor nthreads from dask worker. ( #5414 )
2020-03-16 04:51:24 +08:00
Jiaming Yuan
21b671aa06
[dask] Order the prediction result. ( #5416 )
2020-03-15 19:34:04 +08:00
Jiaming Yuan
ed0216642f
Avoid dask test fixtures. ( #5270 )
...
* Fix Travis OSX timeout.
* Fix classifier.
2020-02-03 12:39:20 +08:00
Jiaming Yuan
ced3660f60
Tests for empty dmatrix. ( #5159 )
2019-12-26 11:51:54 +08:00
Jiaming Yuan
761e938dbe
Support dask dataframe as y for classifier. ( #5077 )
...
* Support dask dataframe as y for classifier.
* Lint.
2019-12-02 11:53:30 +08:00
Jiaming Yuan
7e72a12871
Don't set_params at the end of set_state. ( #4947 )
...
* Don't set_params at the end of set_state.
* Also fix another issue found in dask prediction.
* Add note about prediction.
Don't support other prediction modes at the moment.
2019-10-15 10:08:26 -04:00
Jiaming Yuan
2ebdec8aa6
Fix dask prediction. ( #4941 )
...
* Fix dask prediction.
* Add better error messages for wrong partition.
2019-10-14 23:19:34 -04:00
Jiaming Yuan
6c9b6f11da
Use cudf.concat explicitly. ( #4918 )
...
* Use `cudf.concat` explicitly.
* Add test.
2019-10-10 16:02:10 +13:00
Jiaming Yuan
b8433c455a
Rewrite Dask interface. ( #4819 )
2019-09-25 01:30:14 -04:00
Jiaming Yuan
5374f52531
Complete cudf support. ( #4850 )
...
* Handles missing value.
* Accept all floating point and integer types.
* Move to cudf 9.0 API.
* Remove requirement on `null_count`.
* Arbitrary column types support.
2019-09-16 23:52:00 -04:00
Rory Mitchell
09b90d9329
Add native support for Dask ( #4473 )
...
* Add native support for Dask
* Add multi-GPU demo
* Add sklearn example
2019-05-27 13:29:28 +12:00
