PetrochukM/PyTorch-NLP

Support loading fasttext model from custom file

Open

#61 opened on Jan 3, 2019

View on GitHub
 (5 comments) (0 reactions) (0 assignees)Python (262 forks)batch import
enhancementgood first issuehelp wanted

Repository metrics

Stars
 (2,205 stars)
PR merge metrics
 (No merged PRs in 30d)

Description

What if I want to use own pretrained fasttext model (or even commoncrawl model instead of standard wiki one)? E.g. look what they publish now: https://fasttext.cc/docs/en/crawl-vectors.html. Current FastText impl

    url_base = 'https://s3-us-west-1.amazonaws.com/fasttext-vectors/wiki.{}.vec'
    aligned_url_base = 'https://s3.amazonaws.com/arrival/embeddings/wiki.multi.{}.vec'

    def __init__(self, language="en", aligned=False, **kwargs):
        if aligned:
            url = self.aligned_url_base.format(language)
        else:
url = self.url_base.format(language)

doesn't allow you to do such basic stuff.

Contributor guide