Skip to content

Commit e436896

Browse files
committed
[SPARK-5726] [MLLIB] Remove 'TF' from 'ElementwiseProductTF'
1 parent cb520e6 commit e436896

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ package org.apache.spark.ml.feature
2020
import org.apache.spark.annotation.AlphaComponent
2121
import org.apache.spark.ml.UnaryTransformer
2222
import org.apache.spark.ml.param.{Param, ParamMap}
23-
import org.apache.spark.mllib.feature.ElementwiseProduct
23+
import org.apache.spark.mllib.feature
2424
import org.apache.spark.mllib.linalg.{Vector, VectorUDT}
2525
import org.apache.spark.sql.types.DataType
2626

@@ -29,15 +29,15 @@ import org.apache.spark.sql.types.DataType
2929
* Maps a vector to the hadamard product of it and a reference vector.
3030
*/
3131
@AlphaComponent
32-
class ElementwiseProductTF extends UnaryTransformer[Vector, Vector, ElementwiseProductTF] {
32+
class ElementwiseProduct extends UnaryTransformer[Vector, Vector, ElementwiseProduct] {
3333

3434
/** the vector to multiply with input vectors */
3535
val scalingVec : Param[Vector] = new Param(this, "scalingVector", "vector for hadamard product")
3636
def setScalingVec(value: Vector) = set(scalingVec, value)
3737
def getScalingVec: Vector = get(scalingVec)
3838

3939
override protected def createTransformFunc(paramMap: ParamMap): Vector => Vector = {
40-
val elemScaler = new ElementwiseProduct(paramMap(scalingVec))
40+
val elemScaler = new feature.ElementwiseProduct(paramMap(scalingVec))
4141
elemScaler.transform
4242
}
4343

0 commit comments

Comments
 (0)