diff --git a/tests/unit/test_word_embeddings.py b/tests/unit/test_word_embeddings.py index efa425b..c1eb0e3 100644 --- a/tests/unit/test_word_embeddings.py +++ b/tests/unit/test_word_embeddings.py @@ -15,9 +15,6 @@ from utils_nlp.pretrained_embeddings.glove import ( ) from utils_nlp.pretrained_embeddings.word2vec import ( load_pretrained_vectors as load_word2vec, - load_pretrained_vectors, - download_word2vec_corpus, - extract_word2vec_corpus, ) diff --git a/utils_nlp/pretrained_embeddings/fasttext.py b/utils_nlp/pretrained_embeddings/fasttext.py index 58cda5c..6164013 100644 --- a/utils_nlp/pretrained_embeddings/fasttext.py +++ b/utils_nlp/pretrained_embeddings/fasttext.py @@ -36,6 +36,10 @@ def _download_fasttext_vectors(download_dir, file_name="wiki.simple.zip"): fastText. You can directly download the vectors from here: https://dl.fbaipublicfiles.com/fasttext/vectors-wiki/wiki.simple.zip + For the full version of pre-trained word vectors, change the url for + FASTTEXT_EN_URL to https://dl.fbaipublicfiles.com/fasttext/vectors-wiki/wiki.en.zip + in __init__.py + Args: download_dir (str): File path to download the file file_name (str) : File name given by default but can be changed by the user.