We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2a2983f commit 5eb9100Copy full SHA for 5eb9100
mllib/src/main/scala/org/apache/spark/mllib/feature/Word2Vec.scala
@@ -122,8 +122,13 @@ class Word2Vec extends Serializable with Logging {
122
/** context words from [-window, window] */
123
private val window = 5
124
125
- /** minimum frequency to consider a vocabulary word */
126
- private val minCount = 5
+/** minimum frequency to consider a vocabulary word */
+private var minCount = 5
127
+
128
+def setMinCount(minCount: Int): this.type = {
129
+this.minCount = minCount
130
+this
131
+}
132
133
private var trainWordsCount = 0
134
private var vocabSize = 0
0 commit comments