如何将标签传递给XGBoost JVM包的矩阵

Question

我在官方文档中找到了创建数据矩阵的JVM包快速启动示例

float[] data = new float[] {1f,2f,3f,4f,5f,6f};
int nrow = 3;
int ncol = 2;
float missing = 0.0f;
DMatrix dmat = new DMatrix(data, nrow, ncol, missing);
// and a DMatrix could be called by train method
Booster booster = XGBoost.train(dmat, params, nround, watches, null, null);

然而，文档说JVM包DMatrix的用法与Python包相同，在Python中有一个标签参数

dtrain = xgb.DMatrix(data, label=label)

但是我在JVM包中找不到任何标签，如何设置标签？

Answer 1

在XGBoost JVM包中，我们可以使用dMatrix.setLabel(labelArray)来设置标签(根据XGBoost github发布页面(