diff options
| author | DB Tsai <dbt@netflix.com> | 2015-04-29 21:44:41 -0700 |
|---|---|---|
| committer | Reynold Xin <rxin@databricks.com> | 2015-04-29 21:44:41 -0700 |
| commit | ba49eb1625b1190d8aaf2c55dc1f6309ac3e080c (patch) | |
| tree | efd7ffe7fe29dbd2f41976169529cd99e3765ff5 /mllib | |
| parent | 5553198fe521fb38b600b7687f7780d89a6e1cb9 (diff) | |
| download | spark-ba49eb1625b1190d8aaf2c55dc1f6309ac3e080c.tar.gz spark-ba49eb1625b1190d8aaf2c55dc1f6309ac3e080c.tar.bz2 spark-ba49eb1625b1190d8aaf2c55dc1f6309ac3e080c.zip | |
Some code clean up.
Author: DB Tsai <dbt@netflix.com>
Closes #5794 from dbtsai/clean and squashes the following commits:
ad639dd [DB Tsai] Indentation
834d527 [DB Tsai] Some code clean up.
Diffstat (limited to 'mllib')
| -rw-r--r-- | mllib/src/main/scala/org/apache/spark/mllib/util/LinearDataGenerator.scala | 17 |
1 files changed, 8 insertions, 9 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/util/LinearDataGenerator.scala b/mllib/src/main/scala/org/apache/spark/mllib/util/LinearDataGenerator.scala index 91c2f4ccd3..b1a4517344 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/util/LinearDataGenerator.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/util/LinearDataGenerator.scala @@ -75,8 +75,8 @@ object LinearDataGenerator { seed: Int, eps: Double = 0.1): Seq[LabeledPoint] = { generateLinearInput(intercept, weights, - Array.fill[Double](weights.size)(0.0), - Array.fill[Double](weights.size)(1.0 / 3.0), + Array.fill[Double](weights.length)(0.0), + Array.fill[Double](weights.length)(1.0 / 3.0), nPoints, seed, eps)} /** @@ -105,13 +105,12 @@ object LinearDataGenerator { val x = Array.fill[Array[Double]](nPoints)( Array.fill[Double](weights.length)(rnd.nextDouble())) - x.foreach { - case v => - var i = 0 - while (i < v.length) { - v(i) = (v(i) - 0.5) * math.sqrt(12.0 * xVariance(i)) + xMean(i) - i += 1 - } + x.foreach { v => + var i = 0 + while (i < v.length) { + v(i) = (v(i) - 0.5) * math.sqrt(12.0 * xVariance(i)) + xMean(i) + i += 1 + } } val y = x.map { xi => |