[doc] Add demo for inference using individual tree. (#8752)

2023-02-07 04:40:18 +08:00
parent 28bb01aa22
commit 7b3d473593
3 changed files with 107 additions and 0 deletions
--- a/tests/python/test_demos.py
+++ b/tests/python/test_demos.py
@@ -79,6 +79,12 @@ def test_predict_first_ntree_demo():
    subprocess.check_call(cmd)


+def test_individual_trees():
+    script = os.path.join(PYTHON_DEMO_DIR, 'individual_trees.py')
+    cmd = ['python', script]
+    subprocess.check_call(cmd)
+
+
 def test_predict_leaf_indices_demo():
    script = os.path.join(PYTHON_DEMO_DIR, 'predict_leaf_indices.py')
    cmd = ['python', script]