Commit Graph

1026 Commits

Author SHA1 Message Date
Jiaming Yuan
e8a962575a [EM] Allow staging ellpack on host for GPU external memory. (#10488)
- New parameter `on_host`.
- Abstract format creation and stream creation into policy classes.
2024-06-28 04:42:18 +08:00
david-cortes
61ac8eec8a [R] use Rf_ prefix for R C functions. (#10465) 2024-06-21 14:37:18 +08:00
Jiaming Yuan
43a57c4a85 Bump development version to 2.2. (#10376) 2024-06-04 12:59:16 +08:00
Jiaming Yuan
e6eefea5e2 [coll] Move the rabit poll helper. (#10349) 2024-05-31 08:02:21 +08:00
david-cortes
8998733ef4 [R] Rename BIAS -> (Intercept) (#10337) 2024-05-30 19:43:32 +08:00
david-cortes
949f062229 [R] Fix incorrect division of classification/ranking objectives (#10327) 2024-05-27 10:27:32 +08:00
david-cortes
b2008773bb [R] Update docs for custom user functions (#10328) 2024-05-27 10:26:34 +08:00
david-cortes
5086decb0c [R] Reshape predictions for custom eval metric when they are 2D (#10323) 2024-05-24 17:28:30 +08:00
Jiaming Yuan
a5a58102e5 Revamp the rabit implementation. (#10112)
This PR replaces the original RABIT implementation with a new one, which has already been partially merged into XGBoost. The new one features:
- Federated learning for both CPU and GPU.
- NCCL.
- More data types.
- A unified interface for all the underlying implementations.
- Improved timeout handling for both tracker and workers.
- Exhausted tests with metrics (fixed a couple of bugs along the way).
- A reusable tracker for Python and JVM packages.
2024-05-20 11:56:23 +08:00
Jiaming Yuan
3f64b4fde3 [coll] Add global functions. (#10203) 2024-04-19 03:17:23 +08:00
Jiaming Yuan
4b10200456 [coll] Improve event loop. (#10199)
- Add a test for blocking calls.
- Do not require the queue to be empty after waking up; this frees up the thread to answer blocking calls.
- Handle EOF in read.
- Improve the error message in the result. Allow concatenation of multiple results.
2024-04-18 03:29:52 +08:00
david-cortes
bc9ea62ec0 [R] Make xgb.cv work with xgb.DMatrix only, adding support for survival and ranking fields (#10031)
---------

Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>
2024-03-31 21:53:00 +08:00
Michael Mayer
e0f890ba28 [R] deprecate watchlist (#10110) 2024-03-13 17:02:34 +08:00
david-cortes
b023a253b4 [R] Rename watchlist -> evals (#10032) 2024-03-10 06:48:06 +08:00
Philip Hyunsu Cho
23a37dcaf9 [CI] Test R package with CMake (#10087)
* [CI] Test R package with CMake

* Fix

* Fix

* Update test_r_package.py

* Fix CMake flag for R package

* Install system deps

* Fix

* Use sudo
2024-03-04 12:32:44 -08:00
david-cortes
2c12b956da [R] Refactor callback structure and attributes (#9957) 2024-03-01 15:57:47 +08:00
Jiaming Yuan
3941b31ade Disable column sample by node for the exact tree method. (#10083) 2024-03-01 14:16:10 +08:00
Jiaming Yuan
0ce4372bd4 Use UBJSON for serializing splits for vertical data split. (#10059) 2024-02-25 00:18:23 +08:00
david-cortes
f7005d32c1 [R] Use inplace predict (#9829)
---------

Co-authored-by: Hyunsu Cho <chohyu01@cs.washington.edu>
2024-02-24 02:03:54 +08:00
david-cortes
6e3c899ba7 [R] Don't cap global number of threads for serialization (#10028) 2024-02-20 11:13:00 +08:00
david-cortes
4de866211d [R] switch to URI reader (#10024) 2024-02-05 05:03:38 +08:00
david-cortes
a730c7e67e [R] allow using seed with regular RNG (#10029) 2024-02-04 16:22:22 +08:00
david-cortes
662854c7d7 [R] Document handling of indexes (#10019)
---------

Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
2024-02-02 05:39:09 +08:00
david-cortes
0955213220 [R] rename proxy dmatrix -> data batch (#10016) 2024-01-31 15:43:58 +08:00
david-cortes
1e72dc1276 [R] Add optional check on column names matching in predict (#10020) 2024-01-31 15:43:22 +08:00
david-cortes
5e00a71671 [R] rename slice to avoid dplyr conflict (#10017) 2024-01-31 05:33:37 +08:00
david-cortes
df7cf744b4 [R] Remove enable_categorical parameter (#10018) 2024-01-31 05:17:36 +08:00
david-cortes
3abbbe41ac [R] Add data iterator, quantile dmatrix, external memory, and missing feature_types (#9913) 2024-01-30 19:26:44 +08:00
david-cortes
5062a3ab46 [R] Support booster slicing. (#9948) 2024-01-21 05:11:26 +08:00
david-cortes
c5d0608057 [R] Remove parameters and attributes related to ntree and rebase iterationrange (#9935) 2024-01-21 00:56:57 +08:00
david-cortes
60b9d2eeb9 [R] Avoid memory copies in predict (#9902) 2024-01-21 00:53:18 +08:00
david-cortes
547abb8c12 [R] Remove unusable 'feature_names' argument and make 'model' first argument in inspection functions (#9939) 2024-01-15 17:16:30 +08:00
david-cortes
d3a8d284ab [R] On-demand serialization + standardization of attributes (#9924)
---------

Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
2024-01-11 05:08:42 +08:00
david-cortes
bed0349954 [R] Don't write files to user's directory (#9966) 2024-01-09 03:43:48 +08:00
david-cortes
7ff6d44efa [R] Use R's error stream for printing warnings (#9965) 2024-01-09 03:43:21 +08:00
david-cortes
db396ee340 [R] make sure output fits into int32 (#9949) 2024-01-04 16:51:22 +08:00
david-cortes
3c004a4145 [R] Add missing DMatrix functions (#9929)
* `XGDMatrixGetQuantileCut`
* `XGDMatrixNumNonMissing`
* `XGDMatrixGetDataAsCSR`

---------

Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
2024-01-03 17:29:21 +08:00
david-cortes
49247458f9 [R] Minor improvements for evaluation printing (#9940) 2024-01-03 15:26:55 +08:00
david-cortes
9e33a10202 [R] Replace xgboost() with xgb.train() in most tests and examples (#9941) 2024-01-02 21:20:01 +08:00
david-cortes
32cbab1cc0 [R] put 'verbose' in correct argument (#9942) 2024-01-02 15:20:51 +08:00
david-cortes
73713de601 [R] rename Quality -> Gain (#9938) 2023-12-31 13:01:00 +08:00
david-cortes
8b9c98b65b [R] Clearer function signatures for S3 methods (#9937) 2023-12-31 10:45:04 +08:00
david-cortes
e40c4260ed [R] Enable 'dot' dump format (#9930) 2023-12-30 13:28:27 +08:00
david-cortes
a197899161 [R] avoid leaking exception objects (#9916) 2023-12-26 20:29:55 +08:00
Michael Mayer
52620fdb34 [R] Improve more docstrings (#9919) 2023-12-26 17:30:13 +08:00
Michael Mayer
b807f3e30c [R] improve docstrings for "xgb.Booster.R" (#9906) 2023-12-21 10:01:30 +08:00
david-cortes
ae32936ba2 [R] Catch C++ exceptions (#9903) 2023-12-19 10:45:03 +08:00
david-cortes
ff3d82c006 [R] Refactor field logic for dmatrix (#9901) 2023-12-18 20:31:01 +08:00
Jiaming Yuan
1c6e031c75 [R] Fix clang warning. (#9874) 2023-12-15 01:30:43 +08:00
david-cortes
cd473c9da3 [R] enable multi-dimensional base_margin (#9885) 2023-12-14 09:16:53 +08:00