From fc795e4d7f8792ce4ad70b7ecb41531556669983 Mon Sep 17 00:00:00 2001
From: DB Tsai <dbtsai@alpinenow.com>
Date: Mon, 24 Nov 2014 16:06:22 -0800
Subject: [PATCH] update

---
 .../scala/org/apache/spark/mllib/feature/StandardScaler.scala  | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mllib/src/main/scala/org/apache/spark/mllib/feature/StandardScaler.scala b/mllib/src/main/scala/org/apache/spark/mllib/feature/StandardScaler.scala
index 0343941ce6d46..ac3851b8703fd 100644
--- a/mllib/src/main/scala/org/apache/spark/mllib/feature/StandardScaler.scala
+++ b/mllib/src/main/scala/org/apache/spark/mllib/feature/StandardScaler.scala
@@ -96,7 +96,6 @@ class StandardScalerModel private[mllib] (
    */
   override def transform(vector: Vector): Vector = {
     require(mean.size == vector.size)
-    val localFactor = factor
     if (withMean) {
       val localShift = shift
       vector match {
@@ -104,6 +103,7 @@ class StandardScalerModel private[mllib] (
           val values = dv.values.clone()
           var i = 0
           if(withStd) {
+            val localFactor = factor
             while (i < values.length) {
               values(i) = (values(i) - localShift(i)) * localFactor(i)
               i += 1
@@ -118,6 +118,7 @@ class StandardScalerModel private[mllib] (
         case v => throw new IllegalArgumentException("Do not support vector type " + v.getClass)
       }
     } else if (withStd) {
+      val localFactor = factor
       vector match {
         case dv: DenseVector =>
           val values = dv.values.clone()