enlp.understanding.vectors.word_vectors¶
-
enlp.understanding.vectors.word_vectors(docs, sg=0, mc=0, sz=300, wnd=5, epochs=10)[source]¶ Compute word vectors from corpus
- Parameters
- docs
list list of texts (documents)
- sg
int Training algorithm: 1 for skip-gram; otherwise CBOW.
- mc
int Ignores all words with total frequency lower than this.
- sz
int Dimensionality of the word vectors.
- wnd
int Maximum distance between the current and predicted word within a sentence.
- epochs
int number of epochs for training
- docs
- Returns
- wvs
gensim.model Trained word vector model
- wvs