This commit is contained in:
Casey Hong 2019-06-11 11:45:30 -04:00
Родитель 23d9635230
Коммит e5b12c6f32
2 изменённых файлов: 4 добавлений и 3 удалений

Просмотреть файл

@ -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,
)

Просмотреть файл

@ -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.