Python 在学习排名时，如何在lightgbm中指定预测时的查询分组？

我用

lightgbm

训练了一个学习排序模型，在训练数据集中必须指定

查询大小

，但在预测中我找不到这样的选项，有没有办法传递这个

参数

或者一次只能预测一个查询