Skip to content

Commit 12ed8f9

Browse files
committed
Update Word2Vec.scala
1 parent 76bdf5a commit 12ed8f9

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

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

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -122,14 +122,16 @@ class Word2Vec extends Serializable with Logging {
122122
/** context words from [-window, window] */
123123
private val window = 5
124124

125-
/** The minimum number of times a token must occur in the training corpus to be
126-
* included in the word2vec model (default: 5).
127-
*/
125+
/**
126+
* The minimum number of times a token must occur in the training corpus to be
127+
* included in the word2vec model (default: 5).
128+
*/
128129
private var minCount = 5
129130

130-
/** Sets minCount, the minimum number of times a token must appear to be included in the word2vec model's
131-
* vocabulary (default: 5).
132-
*/
131+
/**
132+
* Sets minCount, the minimum number of times a token must appear to be included in the word2vec
133+
* model's vocabulary (default: 5).
134+
*/
133135
def setMinCount(minCount: Int): this.type = {
134136
this.minCount = minCount
135137
this

0 commit comments

Comments
 (0)