Skip to content

Commit

Permalink
Update Word2Vec.scala
Browse files Browse the repository at this point in the history
Added javadoc
  • Loading branch information
ganonp committed Dec 16, 2014
1 parent 5eb9100 commit ffb88bb
Showing 1 changed file with 11 additions and 8 deletions.
19 changes: 11 additions & 8 deletions mllib/src/main/scala/org/apache/spark/mllib/feature/Word2Vec.scala
Original file line number Diff line number Diff line change
Expand Up @@ -121,14 +121,17 @@ class Word2Vec extends Serializable with Logging {

/** context words from [-window, window] */
private val window = 5

/** minimum frequency to consider a vocabulary word */
private var minCount = 5

def setMinCount(minCount: Int): this.type = {
this.minCount = minCount
this
}

/** minimum frequency to consider a vocabulary word */
private var minCount = 5

/** Sets the minimum frequency a token must appear to be included in the word2vec model's
* vocabulary (default: 5).
*/
def setMinCount(minCount: Int): this.type = {
this.minCount = minCount
this
}

private var trainWordsCount = 0
private var vocabSize = 0
Expand Down

0 comments on commit ffb88bb

Please sign in to comment.