dmlc / dmlc/xgboost

Cannot retrieve 'qid' field from DMatrix

Open
#9,861 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
C++
Stars
28.8k
Forks
8.9k
Avg merge
1d 12h
Merged PRs (30d)
54

Description

It doesn't seem to be possible to retrieve the 'qid' field of a DMatrix, neither as an integer field, nor as a float field.

Example:
```python
import numpy as np, xgboost as xgb
rng = np.random.default_rng(seed=13)
X = rng.standard_normal(size=(100,10))
y = rng.integers(1, size=100)
qid = rng.integers(3, size=100)
qid = qid[np.argsort(qid)]
dm = xgb.DMatrix(X, label=y, qid=qid)
```
```python
dm.get_uint_info("qid")
```
```
XGBoostError: [20:23:13] [/workspace/src/data/data.cc:628](https://untitled+.vscode-resource.vscode-cdn.net/workspace/src/data/data.cc:628): Unknown uint32 field name: qid
```
```python
dm.get_float_info("qid")
```
```
XGBoostError: [20:23:26] [/workspace/src/data/data.cc:619](https://untitled+.vscode-resource.vscode-cdn.net/workspace/src/data/data.cc:619): Unknown float field name: qid
```

Since one can get the group when supplying `qid`, I guess it should be possible to retrieve `qid` too.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.