spaCy/spacy/en/download.py

26 lines
694 B
Python
Raw Normal View History

2015-11-15 17:58:21 +03:00
import plac
2016-10-23 20:46:28 +03:00
import sputnik
from ..download import download
2016-10-23 20:46:28 +03:00
from .. import about
@plac.annotations(
force=("Force overwrite", "flag", "f", bool),
data_path=("Path to download model", "option", "d", str)
)
def main(data_size='all', force=False, data_path=None):
2016-10-23 20:46:28 +03:00
if force:
sputnik.purge(about.__title__, about.__version__)
2016-10-19 01:52:16 +03:00
if data_size in ('all', 'parser'):
print("Downloading parsing model")
download('en', force=False, data_path=data_path)
2016-10-19 01:52:16 +03:00
if data_size in ('all', 'glove'):
print("Downloading GloVe vectors")
download('en_glove_cc_300_1m_vectors', force=False, data_path=data_path)
2015-01-02 13:44:41 +03:00
if __name__ == '__main__':
2015-01-31 05:51:56 +03:00
plac.call(main)