From 9b465052cef6676090ff1dc2f9f8642a71d3bbc9 Mon Sep 17 00:00:00 2001 From: Bobby Wang Date: Thu, 9 May 2024 07:44:07 +0800 Subject: [PATCH] [jvm-packages] fix group col for gpu packages (#10254) --- .../ml/dmlc/xgboost4j/scala/rapids/spark/GpuPreXGBoost.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jvm-packages/xgboost4j-spark-gpu/src/main/scala/ml/dmlc/xgboost4j/scala/rapids/spark/GpuPreXGBoost.scala b/jvm-packages/xgboost4j-spark-gpu/src/main/scala/ml/dmlc/xgboost4j/scala/rapids/spark/GpuPreXGBoost.scala index d34802805..7e83dc6f1 100644 --- a/jvm-packages/xgboost4j-spark-gpu/src/main/scala/ml/dmlc/xgboost4j/scala/rapids/spark/GpuPreXGBoost.scala +++ b/jvm-packages/xgboost4j-spark-gpu/src/main/scala/ml/dmlc/xgboost4j/scala/rapids/spark/GpuPreXGBoost.scala @@ -160,7 +160,7 @@ object GpuPreXGBoost extends PreXGBoostProvider { // Check columns and build column data batch val trainingData = GpuUtils.buildColumnDataBatch(feturesCols, - labelName, weightName, marginName, "", castedDF) + labelName, weightName, marginName, groupName, castedDF) // eval map val evalDataMap = evalSets.map {