Skip to content

Commit 76bdf5a

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

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

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

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

125-
/** minimum frequency to consider a vocabulary word */
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+
*/
126128
private var minCount = 5
127129

128-
/** Sets the minimum frequency a token must appear to be included in the word2vec model's
130+
/** Sets minCount, the minimum number of times a token must appear to be included in the word2vec model's
129131
* vocabulary (default: 5).
130132
*/
131133
def setMinCount(minCount: Int): this.type = {

0 commit comments

Comments
 (0)