machine-learning - 如何为 XGBoost XGBRanker 准备特征和标签？

Question

我将使用 XGBRanker 制作推荐系统。但是在the official docs我还没有找到如何准备数据集的示例。

那么，XGBRanker 中应该采用哪种格式的特征和标签？

Answer 1

XGBRanker 我们需要以下数据集:

1. X 作为特征
2. Y 作为目标值(标签)
3. group as kwargs(根据域定义的唯一组)/每行 qid

请注意，组信息应该是一个数组，由每个组的元素数组成。

例如:

Groups in data: 4

Training, Test: (100,50)

Training Data(X,Y): (100,3)[feature 1, feature 2, feature 3]

Training Label(Y): (100,1)

Training group[A,B,C,D] i.e **grpInfo**: [25,25,10,35]

注意:grpInfo 的总和应等于 TrainingSet 中的记录数，即在本例中为 100

XGBRanker(X[[特征 1, 特征 2, 特征 3]],Y[[标签]],group=grpInfo)