@@ -20,7 +20,7 @@ package org.apache.spark.ml.feature
2020import org .apache .spark .annotation .AlphaComponent
2121import org .apache .spark .ml .UnaryTransformer
2222import org .apache .spark .ml .param .{Param , ParamMap }
23- import org .apache .spark .mllib .feature . ElementwiseProduct
23+ import org .apache .spark .mllib .feature
2424import org .apache .spark .mllib .linalg .{Vector , VectorUDT }
2525import 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