Add kwargs to Corpus.dev_dataset to match train_dataset

This commit is contained in:
Matthew Honnibal 2020-06-21 17:16:55 +02:00
parent 59098a5f62
commit 7d329cd1ac

View File

@ -76,7 +76,7 @@ class Corpus:
random.shuffle(examples) random.shuffle(examples)
yield from 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.train_loc)) ref_docs = self.read_docbin(nlp.vocab, self.walk_corpus(self.train_loc))
examples = self.make_examples(nlp, ref_docs, **kwargs) examples = self.make_examples(nlp, ref_docs, **kwargs)
yield from examples yield from examples