Skip to content

Commit fac12ad

Browse files
committed
[SPARK-5726] [MLLIB] Use new createTransformFunc.
1 parent 90f7e39 commit fac12ad

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

mllib/src/main/scala/org/apache/spark/ml/feature/ElementwiseProduct.scala

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ package org.apache.spark.ml.feature
1919

2020
import org.apache.spark.annotation.AlphaComponent
2121
import org.apache.spark.ml.UnaryTransformer
22-
import org.apache.spark.ml.param.{Param, ParamMap}
22+
import org.apache.spark.ml.param.Param
2323
import org.apache.spark.mllib.feature
2424
import org.apache.spark.mllib.linalg.{Vector, VectorUDT}
2525
import org.apache.spark.sql.types.DataType
@@ -45,9 +45,9 @@ class ElementwiseProduct extends UnaryTransformer[Vector, Vector, ElementwisePro
4545
/** @group getParam */
4646
def getScalingVec: Vector = getOrDefault(scalingVec)
4747

48-
override protected def createTransformFunc(paramMap: ParamMap): Vector => Vector = {
49-
require(paramMap.contains(scalingVec), s"transformation requires a weight vector: $scalingVec")
50-
val elemScaler = new feature.ElementwiseProduct(paramMap(scalingVec))
48+
override protected def createTransformFunc: Vector => Vector = {
49+
require(params.contains(scalingVec), s"transformation requires a weight vector")
50+
val elemScaler = new feature.ElementwiseProduct($(scalingVec))
5151
elemScaler.transform
5252
}
5353

0 commit comments

Comments
 (0)