A python wrapper for embedding short texts or sentences using sent2vec, which draws on FastText.
To embed a list of strings documents
, use:
from nk_sent2vec import Sent2Vec
vectorizer = Sent2Vec(path = '/root/models/torontobooks_unigrams.bin')
print(vectorizer.embed_sentences(sentences=documents))
Tests can be run using pytest -s tests
Also see makefile
for default commands