mirror of
https://github.com/explosion/spaCy.git
synced 2024-11-14 05:37:03 +03:00
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
from os import path
|
|
import os
|
|
import tarfile
|
|
import shutil
|
|
import requests
|
|
|
|
PARSER_URL = 'http://s3-us-west-1.amazonaws.com/media.spacynlp.com/en.tgz'
|
|
|
|
DEP_VECTORS_URL = 'http://u.cs.biu.ac.il/~yogo/data/syntemb/deps.words.bz2'
|
|
|
|
DEST_DIR = path.join(path.dirname(__file__), 'data')
|
|
|
|
def download_file(url):
|
|
local_filename = url.split('/')[-1]
|
|
return path.join(DEST_DIR, local_filename)
|
|
# NOTE the stream=True parameter
|
|
r = requests.get(url, stream=True)
|
|
print "Download %s" % url
|
|
i = 0
|
|
with open(local_filename, 'wb') as f:
|
|
for chunk in r.iter_content(chunk_size=1024):
|
|
if chunk: # filter out keep-alive new chunks
|
|
f.write(chunk)
|
|
f.flush()
|
|
print i
|
|
i += 1
|
|
return local_filename
|
|
|
|
def install_parser_model(url, dest_dir):
|
|
if not os.path.exists(dest_dir):
|
|
os.mkdir(dest_dir)
|
|
assert not path.exists(path.join(dest_dir, 'en'))
|
|
|
|
filename = download_file(url)
|
|
t = tarfile.open(filename, mode=":gz")
|
|
t.extractall(dest_dir)
|
|
shutil.move(path.join(dest_dir, 'en', 'deps', 'model'), dest_dir)
|
|
shutil.move(path.join(dest_dir, 'en', 'deps', 'config.json'), dest_dir)
|
|
shutil.rmtree(path.join(dest_dir, 'en'))
|
|
|
|
|
|
def install_dep_vectors(url, dest_dir):
|
|
if not os.path.exists(dest_dir):
|
|
os.mkdir(dest_dir)
|
|
|
|
filename = download_file(url)
|
|
shutil.move(filename, path.join(dest_dir, 'vec.bz2'))
|
|
|
|
|
|
def main():
|
|
#install_parser_model(PARSER_URL, path.join(DEST_DIR, 'deps'))
|
|
install_dep_vectors(DEP_VECTORS_URL, path.join(DEST_DIR, 'vocab'))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|