diff --git a/mllib/src/main/scala/org/apache/spark/ml/feature/StandardScaler.scala b/mllib/src/main/scala/org/apache/spark/ml/feature/StandardScaler.scala index 2ce94acea694c..a30fbc383254d 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/feature/StandardScaler.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/feature/StandardScaler.scala @@ -40,7 +40,7 @@ private[feature] trait StandardScalerParams extends Params with HasInputCol with ) } -setDefault(withMean -> false, withStd -> true) + /** * :: AlphaComponent :: @@ -50,6 +50,8 @@ setDefault(withMean -> false, withStd -> true) @AlphaComponent class StandardScaler extends Estimator[StandardScalerModel] with StandardScalerParams { + setDefault(withMean -> false, withStd -> true) + /** @group setParam */ def setInputCol(value: String): this.type = set(inputCol, value)