Skip to content

Commit 5dd4ee7

Browse files
committed
fix scala style
1 parent 743e0d5 commit 5dd4ee7

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

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

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)