mirror of
https://github.com/explosion/spaCy.git
synced 2024-12-26 01:46:28 +03:00
Merge pull request #286 from gushecht/patch-2
added batch_size as keyword argument
This commit is contained in:
commit
b37571063a
|
@ -58,7 +58,7 @@ def save_parses(batch_id, input_, out_dir, n_threads, batch_size):
|
||||||
n_thread=("Number of threads per process", "option", "t", int),
|
n_thread=("Number of threads per process", "option", "t", int),
|
||||||
batch_size=("Number of texts to accumulate in a buffer", "option", "b", int)
|
batch_size=("Number of texts to accumulate in a buffer", "option", "b", int)
|
||||||
)
|
)
|
||||||
def main(in_loc, out_dir, n_process=1, n_thread=4):
|
def main(in_loc, out_dir, n_process=1, n_thread=4, batch_size=100):
|
||||||
if not path.exists(out_dir):
|
if not path.exists(out_dir):
|
||||||
path.join(out_dir)
|
path.join(out_dir)
|
||||||
if n_process >= 2:
|
if n_process >= 2:
|
||||||
|
|
Loading…
Reference in New Issue
Block a user