[doc] Update for prediction. (#7648)

2022-02-15 05:01:55 +08:00
parent 0d0abe1845
commit 0da7d872ef
2 changed files with 5 additions and 3 deletions
--- a/demo/guide-python/predict_leaf_indices.py
+++ b/demo/guide-python/predict_leaf_indices.py
@@ -16,7 +16,9 @@ bst = xgb.train(param, dtrain, num_round, watchlist)

 print('start testing predict the leaf indices')
 # predict using first 2 tree
-leafindex = bst.predict(dtest, ntree_limit=2, pred_leaf=True)
+leafindex = bst.predict(
+    dtest, iteration_range=(0, 2), pred_leaf=True, strict_shape=True
+)
 print(leafindex.shape)
 print(leafindex)
 # predict all trees