Skip to content

Commit 9b46505

Browse files
authored
[jvm-packages] fix group col for gpu packages (dmlc#10254)
1 parent 8237920 commit 9b46505

File tree

1 file changed

+1
-1
lines changed
  • jvm-packages/xgboost4j-spark-gpu/src/main/scala/ml/dmlc/xgboost4j/scala/rapids/spark

1 file changed

+1
-1
lines changed

jvm-packages/xgboost4j-spark-gpu/src/main/scala/ml/dmlc/xgboost4j/scala/rapids/spark/GpuPreXGBoost.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ object GpuPreXGBoost extends PreXGBoostProvider {
160160

161161
// Check columns and build column data batch
162162
val trainingData = GpuUtils.buildColumnDataBatch(feturesCols,
163-
labelName, weightName, marginName, "", castedDF)
163+
labelName, weightName, marginName, groupName, castedDF)
164164

165165
// eval map
166166
val evalDataMap = evalSets.map {

0 commit comments

Comments
 (0)