diff --git a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoost.scala b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoost.scala index 78e844283..9fe5fd264 100644 --- a/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoost.scala +++ b/jvm-packages/xgboost4j-spark/src/main/scala/ml/dmlc/xgboost4j/scala/spark/XGBoost.scala @@ -48,7 +48,7 @@ object XGBoost extends Serializable { val indices = new ListBuffer[Int] val values = new ListBuffer[Double] for (i <- dVector.values.indices) { - if (values(i) != missing) { + if (dVector.values(i) != missing) { indices += i values += dVector.values(i) }