explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-10 08:12:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix Corpus

Browse Source
This commit is contained in:
Matthew Honnibal 2020-06-22 16:28:47 +02:00
parent fedfabec80
commit afe6ee4548
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/gold/corpus.py
View File

@ -72,7 +72,7 @@ class Corpus:
i += 1
return n
def train_dataset(self, nlp, shuffle=True):
def train_dataset(self, nlp, shuffle=True, **kwargs):
ref_docs = self.read_docbin(nlp.vocab, self.walk_corpus(self.train_loc))
examples = self.make_examples(nlp, ref_docs)
if shuffle:
@ -80,7 +80,7 @@ class Corpus:
random.shuffle(examples)
yield from examples
def dev_dataset(self, nlp):
def dev_dataset(self, nlp, **kwargs):
ref_docs = self.read_docbin(nlp.vocab, self.walk_corpus(self.dev_loc))
examples = self.make_examples(nlp, ref_docs)
yield from examples