add require for one-based in loadLIBSVM

nemccarthy · May 31, 2015 · 5bd1f9a · 5bd1f9a
1 parent 97a60cf
commit 5bd1f9a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/util/MLUtils.scala b/mllib/src/main/scala/org/apache/spark/mllib/util/MLUtils.scala
@@ -82,6 +82,8 @@ object MLUtils {
           val value = indexAndValue(1).toDouble
           (index, value)
         }.unzip
+        require(indices.size == 0 || indices(0) >= 0,
+          "indices should be one-based in LIBSVM format")
         (label, indices.toArray, values.toArray)
       }