Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
xgboost/tests/python
History
Jiaming Yuan 80065d571e
[dask] Add DaskXGBRanker (#6576) 
* Initial support for distributed LTR using dask.

* Support `qid` in libxgboost.
* Refactor `predict` and `n_features_in_`, `best_[score/iteration/ntree_limit]`
  to avoid duplicated code.
* Define `DaskXGBRanker`.

The dask ranker doesn't support group structure, instead it uses query id and
convert to group ptr internally.
2021-01-08 18:35:09 +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
Support early stopping with training continuation, correct num boosted rounds. (#6506)
2020-12-17 19:59:19 +08:00
test_basic.py
Move metric configuration into booster. (#6504)
2020-12-16 05:35:04 +08:00
test_callback.py
Support early stopping with training continuation, correct num boosted rounds. (#6506)
2020-12-17 19:59:19 +08:00
test_cli.py
Fix CLI ranking demo. (#6439)
2020-11-29 03:12:06 +08:00
test_config.py
Add global configuration (#6414)
2020-12-03 00:05:18 -08:00
test_demos.py
Group CLI demo into subdirectory. (#6258)
2020-10-28 14:40:44 -07:00
test_dmatrix.py
[dask] Add DaskXGBRanker (#6576)
2021-01-08 18:35:09 +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 pytest conventions consistently (#6337)
2020-11-19 17:00:15 -08:00
test_ranking.py
[dask] Add DaskXGBRanker (#6576)
2021-01-08 18:35:09 +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
Fix best_ntree_limit. (#6569)
2021-01-03 05:58:54 +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
[dask] Add DaskXGBRanker (#6576)
2021-01-08 18:35:09 +08:00
test_with_modin.py
Use pytest conventions consistently (#6337)
2020-11-19 17:00:15 -08:00
test_with_pandas.py
Use pytest conventions consistently (#6337)
2020-11-19 17:00:15 -08:00
test_with_shap.py
Update GPUTreeshap (#6163)
2020-09-28 09:43:47 +13:00
test_with_sklearn.py
[dask] Add DaskXGBRanker (#6576)
2021-01-08 18:35:09 +08:00
testing.py
[dask] Add DaskXGBRanker (#6576)
2021-01-08 18:35:09 +08:00