Update SVM.scala

apache · Mar 17, 2015 · 158a766 · 158a766
1 parent e3f315a
commit 158a766
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/classification/SVM.scala b/mllib/src/main/scala/org/apache/spark/mllib/classification/SVM.scala
@@ -128,6 +128,7 @@ class SVMWithSGD private (
     private var miniBatchFraction: Double)
   extends GeneralizedLinearAlgorithm[SVMModel] with Serializable {
 
+  this.setFeatureScaling(true)
   private val gradient = new HingeGradient()
   private val updater = new SquaredL2Updater()
   override val optimizer = new GradientDescent(gradient, updater)