File tree Expand file tree Collapse file tree 1 file changed +11
-8
lines changed
mllib/src/main/scala/org/apache/spark/mllib/feature Expand file tree Collapse file tree 1 file changed +11
-8
lines changed Original file line number Diff line number Diff line change @@ -121,14 +121,17 @@ class Word2Vec extends Serializable with Logging {
121
121
122
122
/** context words from [-window, window] */
123
123
private val window = 5
124
-
125
- /** minimum frequency to consider a vocabulary word */
126
- private var minCount = 5
127
-
128
- def setMinCount (minCount : Int ): this .type = {
129
- this .minCount = minCount
130
- this
131
- }
124
+
125
+ /** minimum frequency to consider a vocabulary word */
126
+ private var minCount = 5
127
+
128
+ /** Sets the minimum frequency a token must appear to be included in the word2vec model's
129
+ * vocabulary (default: 5).
130
+ */
131
+ def setMinCount (minCount : Int ): this .type = {
132
+ this .minCount = minCount
133
+ this
134
+ }
132
135
133
136
private var trainWordsCount = 0
134
137
private var vocabSize = 0
You can’t perform that action at this time.
0 commit comments