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

[jvm-packages] IndexOutOfBoundsException (#1589)

Browse Source 
ml.dmlc.xgboost4j.scala.spark.XGBoost.scala:51

values is empty when we meet it at first time, so values(0) throw an IndexOutOfBoundsException.
It should be  dVector.values(i) instead of values(i).
This commit is contained in:
reg.zhuce 2016-09-20 21:13:47 +08:00 committed by Nan Zhu
parent d8876b0b73
commit 3ee145b8dc
1 changed files with 1 additions and 1 deletions
Show Stats 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

@ -48,7 +48,7 @@ object XGBoost extends Serializable {
val indices = new ListBuffer[Int] val indices = new ListBuffer[Int]
val values = new ListBuffer[Double] val values = new ListBuffer[Double]
for (i <- dVector.values.indices) { for (i <- dVector.values.indices) {
if (values(i) != missing) { if (dVector.values(i) != missing) {
indices += i indices += i
values += dVector.values(i) values += dVector.values(i)
} }