Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[jvm-packages] For training data with group, empty RDD partition threw exception (#3749) (#3750)

Browse Source
This commit is contained in:
weitian
2018-10-09 09:03:22 -07:00
committed by Nan Zhu
parent ca33bf6476
commit 9504f411c1
3 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoost.scala
View File

@@ -497,7 +497,7 @@ private[spark] class LabeledPointGroupIterator(base: Iterator[XGBLabeledPoint])
extends AbstractIterator[XGBLabeledPointGroup] {
private var firstPointOfNextGroup: XGBLabeledPoint = null
private var isNewGroup = true
private var isNewGroup = false
override def hasNext: Boolean = {
return base.hasNext || isNewGroup