Update documents and tests. (#7659)

* Revise documents after recent refactoring and cat support. * Add tests for behavior of max_depth and max_leaves.
2022-02-26 03:57:47 +08:00
parent 5eed2990ad
commit 18a4af63aa
7 changed files with 142 additions and 44 deletions
--- a/tests/python-gpu/test_gpu_prediction.py
+++ b/tests/python-gpu/test_gpu_prediction.py
@@ -22,7 +22,7 @@ from test_predict import run_predict_leaf      # noqa
 rng = np.random.RandomState(1994)

 shap_parameter_strategy = strategies.fixed_dictionaries({
-    'max_depth': strategies.integers(0, 11),
+    'max_depth': strategies.integers(1, 11),
    'max_leaves': strategies.integers(0, 256),
    'num_parallel_tree': strategies.sampled_from([1, 10]),
 }).filter(lambda x: x['max_depth'] > 0 or x['max_leaves'] > 0)