aboutsummaryrefslogtreecommitdiff
path: root/mllib
diff options
context:
space:
mode:
authorXiangrui Meng <meng@databricks.com>2015-10-20 18:37:29 -0700
committerXiangrui Meng <meng@databricks.com>2015-10-20 18:37:29 -0700
commit45861693bef2619196f0fbdf5c166ad3f9b1e8d1 (patch)
tree3442d88c993526b709e30151da59a59b1beba3b7 /mllib
parent89e6db6150704deab46232352d1986bc1449883b (diff)
downloadspark-45861693bef2619196f0fbdf5c166ad3f9b1e8d1.tar.gz
spark-45861693bef2619196f0fbdf5c166ad3f9b1e8d1.tar.bz2
spark-45861693bef2619196f0fbdf5c166ad3f9b1e8d1.zip
[SPARK-10082][MLLIB] minor style updates for matrix indexing after #8271
* `>=0` => `>= 0` * print `i`, `j` in the log message MechCoder Author: Xiangrui Meng <meng@databricks.com> Closes #9189 from mengxr/SPARK-10082.
Diffstat (limited to 'mllib')
-rw-r--r--mllib/src/main/scala/org/apache/spark/mllib/linalg/Matrices.scala8
-rw-r--r--mllib/src/test/scala/org/apache/spark/mllib/linalg/MatricesSuite.scala8
2 files changed, 8 insertions, 8 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/linalg/Matrices.scala b/mllib/src/main/scala/org/apache/spark/mllib/linalg/Matrices.scala
index cfed9ad073..8ba6e4e78d 100644
--- a/mllib/src/main/scala/org/apache/spark/mllib/linalg/Matrices.scala
+++ b/mllib/src/main/scala/org/apache/spark/mllib/linalg/Matrices.scala
@@ -296,8 +296,8 @@ class DenseMatrix @Since("1.3.0") (
override def apply(i: Int, j: Int): Double = values(index(i, j))
private[mllib] def index(i: Int, j: Int): Int = {
- require(i < numRows && i >=0, s"Expected 0 <= i < $numRows, got $i")
- require(j < numCols && j >=0, s"Expected 0 <= j < $numCols, got $j")
+ require(i >= 0 && i < numRows, s"Expected 0 <= i < $numRows, got i = $i.")
+ require(j >= 0 && j < numCols, s"Expected 0 <= j < $numCols, got j = $j.")
if (!isTransposed) i + numRows * j else j + numCols * i
}
@@ -572,8 +572,8 @@ class SparseMatrix @Since("1.3.0") (
}
private[mllib] def index(i: Int, j: Int): Int = {
- require(i < numRows && i >=0, s"Expected 0 <= i < $numRows, got $i")
- require(j < numCols && j >=0, s"Expected 0 <= j < $numCols, got $j")
+ require(i >= 0 && i < numRows, s"Expected 0 <= i < $numRows, got i = $i.")
+ require(j >= 0 && j < numCols, s"Expected 0 <= j < $numCols, got j = $j.")
if (!isTransposed) {
Arrays.binarySearch(rowIndices, colPtrs(j), colPtrs(j + 1), i)
} else {
diff --git a/mllib/src/test/scala/org/apache/spark/mllib/linalg/MatricesSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/linalg/MatricesSuite.scala
index b0071c9a02..1833cf3833 100644
--- a/mllib/src/test/scala/org/apache/spark/mllib/linalg/MatricesSuite.scala
+++ b/mllib/src/test/scala/org/apache/spark/mllib/linalg/MatricesSuite.scala
@@ -78,10 +78,10 @@ class MatricesSuite extends SparkFunSuite {
val sm = Matrices.sparse(3, 2, Array(0, 2, 3), Array(1, 2, 1), Array(0.0, 1.0, 2.0))
val dm = Matrices.dense(3, 2, Array(0.0, 2.3, 1.4, 3.2, 1.0, 9.1))
Array(sm, dm).foreach { mat =>
- intercept[IllegalArgumentException] { mat.index(4, 1) }
- intercept[IllegalArgumentException] { mat.index(1, 4) }
- intercept[IllegalArgumentException] { mat.index(-1, 2) }
- intercept[IllegalArgumentException] { mat.index(1, -2) }
+ intercept[IllegalArgumentException] { mat.index(4, 1) }
+ intercept[IllegalArgumentException] { mat.index(1, 4) }
+ intercept[IllegalArgumentException] { mat.index(-1, 2) }
+ intercept[IllegalArgumentException] { mat.index(1, -2) }
}
}