Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[doc] Update learning_to_rank.rst (#9336)

Browse Source
This commit is contained in:
jasjung
2023-06-26 22:56:18 -07:00
committed by GitHub
parent cfa9c42eb4
commit 96c3071a8a
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
doc/tutorials/learning_to_rank.rst
View File

@@ -48,8 +48,9 @@ Notice that the samples are sorted based on their query index in a non-decreasin
import xgboost as xgb import xgboost as xgb
# Make a synthetic ranking dataset for demonstration # Make a synthetic ranking dataset for demonstration
X, y = make_classification(random_state=rng) seed = 1994
rng = np.random.default_rng(1994) X, y = make_classification(random_state=seed)
rng = np.random.default_rng(seed)
n_query_groups = 3 n_query_groups = 3
qid = rng.integers(0, 3, size=X.shape[0]) qid = rng.integers(0, 3, size=X.shape[0])