@@ -106,22 +106,22 @@ final class Word2Vec extends Estimator[Word2VecModel] with Word2VecBase {
106106 def setOutputCol (value : String ): this .type = set(outputCol, value)
107107
108108 /** @group setParam */
109- def setVectorSize (value : Int ) = set(vectorSize, value)
109+ def setVectorSize (value : Int ): this . type = set(vectorSize, value)
110110
111111 /** @group setParam */
112- def setStepSize (value : Double ) = set(stepSize, value)
112+ def setStepSize (value : Double ): this . type = set(stepSize, value)
113113
114114 /** @group setParam */
115- def setNumPartitions (value : Int ) = set(numPartitions, value)
115+ def setNumPartitions (value : Int ): this . type = set(numPartitions, value)
116116
117117 /** @group setParam */
118- def setMaxIter (value : Int ) = set(maxIter, value)
118+ def setMaxIter (value : Int ): this . type = set(maxIter, value)
119119
120120 /** @group setParam */
121- def setSeed (value : Long ) = set(seed, value)
121+ def setSeed (value : Long ): this . type = set(seed, value)
122122
123123 /** @group setParam */
124- def setMinCount (value : Int ) = set(minCount, value)
124+ def setMinCount (value : Int ): this . type = set(minCount, value)
125125
126126 override def fit (dataset : DataFrame , paramMap : ParamMap ): Word2VecModel = {
127127 transformSchema(dataset.schema, paramMap, logging = true )
@@ -159,7 +159,7 @@ class Word2VecModel private[ml] (
159159 /** @group setParam */
160160 def setInputCol (value : String ): this .type = set(inputCol, value)
161161
162- /** @group setParam */
162+ /** @group setParam */
163163 def setOutputCol (value : String ): this .type = set(outputCol, value)
164164
165165 /**
0 commit comments