Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
xgboost/tests/python
History
Philip Hyunsu Cho 366f3cb9d8
Add use_rmm flag to global configuration (#6656) 
* Ensure RMM is 0.18 or later

* Add use_rmm flag to global configuration

* Modify XGBCachingDeviceAllocatorImpl to skip CUB when use_rmm=True

* Update the demo

* [CI] Pin NumPy to 1.19.4, since NumPy 1.19.5 doesn't work with latest Shap
2021-03-09 14:53:05 -08:00
..
generate_models.py
Enable loading model from <1.0.0 trained with objective='binary:logitraw' (#6517)
2020-12-16 16:53:46 -08:00
test_basic_models.py
[breaking] Save booster feature info in JSON, remove feature name generation. (#6605)
2021-02-25 18:54:16 +08:00
test_basic.py
Move metric configuration into booster. (#6504)
2020-12-16 05:35:04 +08:00
test_callback.py
Fix learning rate scheduler with cv. (#6720)
2021-02-28 13:57:42 +08:00
test_cli.py
[breaking] Save booster feature info in JSON, remove feature name generation. (#6605)
2021-02-25 18:54:16 +08:00
test_config.py
Add use_rmm flag to global configuration (#6656)
2021-03-09 14:53:05 -08:00
test_demos.py
Group CLI demo into subdirectory. (#6258)
2020-10-28 14:40:44 -07:00
test_dmatrix.py
[breaking] Save booster feature info in JSON, remove feature name generation. (#6605)
2021-02-25 18:54:16 +08:00
test_dt.py
Use pytest conventions consistently (#6337)
2020-11-19 17:00:15 -08:00
test_early_stopping.py
Remove warnings in tests. (#6554)
2020-12-31 13:41:18 +08:00
test_eval_metrics.py
Remove warnings in tests. (#6554)
2020-12-31 13:41:18 +08:00
test_interaction_constraints.py
Use pytest conventions consistently (#6337)
2020-11-19 17:00:15 -08:00
test_linear.py
Relax linear test. (#5849)
2020-07-03 07:49:53 +08:00
test_model_compatibility.py
Enable loading model from <1.0.0 trained with objective='binary:logitraw' (#6517)
2020-12-16 16:53:46 -08:00
test_monotone_constraints.py
Use pytest conventions consistently (#6337)
2020-11-19 17:00:15 -08:00
test_openmp.py
Use pytest conventions consistently (#6337)
2020-11-19 17:00:15 -08:00
test_parse_tree.py
Use pytest conventions consistently (#6337)
2020-11-19 17:00:15 -08:00
test_pickling.py
Drop saving binary format for memory snapshot. (#6513)
2020-12-17 00:14:57 +08:00
test_plotting.py
Use pytest conventions consistently (#6337)
2020-11-19 17:00:15 -08:00
test_predict.py
Use Predictor for dart. (#6693)
2021-02-09 23:30:19 +08:00
test_ranking.py
Remove duplicated DMatrix. (#6592)
2021-01-12 09:36:56 +08:00
test_shap.py
Use pytest conventions consistently (#6337)
2020-11-19 17:00:15 -08:00
test_survival.py
[dask] Allow empty data matrix in AFT survival (#6379)
2020-11-12 17:49:58 -08:00
test_tracker.py
[dask] Supoort running on GKE. (#6343)
2020-11-11 18:04:34 +08:00
test_training_continuation.py
Revert ntree limit fix (#6616)
2021-01-19 23:51:16 +08:00
test_tree_regularization.py
Use pytest conventions consistently (#6337)
2020-11-19 17:00:15 -08:00
test_updaters.py
Use pytest conventions consistently (#6337)
2020-11-19 17:00:15 -08:00
test_with_dask.py
Add use_rmm flag to global configuration (#6656)
2021-03-09 14:53:05 -08:00
test_with_modin.py
Use pytest conventions consistently (#6337)
2020-11-19 17:00:15 -08:00
test_with_pandas.py
Fix DMatrix slice with feature types. (#6689)
2021-02-09 08:13:51 +08:00
test_with_shap.py
Enhance inplace prediction. (#6653)
2021-02-02 11:41:46 +08:00
test_with_sklearn.py
Fix divide by 0 in feature importance when no split is found. (#6676)
2021-02-05 03:39:30 +08:00
testing.py
[dask] Add DaskXGBRanker (#6576)
2021-01-08 18:35:09 +08:00