We are trying to train a binary classification (0/1) model using XGBoost4J-Spark and it is getting stuck at
foreachPartition at XGBoost.scala:565. The model training is working for different datasets of similar size, but for a particular dataset with the dependent variable distribution of
0: 49.09% 1: 50.91%
the training gets stuck at
foreachPartition at XGBoost.scala:565.