Fix size limits in training data

This commit is contained in:
Matthew Honnibal 2018-05-15 19:01:41 +02:00
parent 8446b35ce0
commit f328c195ca

View File

@ -160,7 +160,7 @@ class GoldCorpus(object):
yield item
i += len(item[1])
if limit and i >= limit:
break
return
@property
def dev_tuples(self):
@ -180,7 +180,7 @@ class GoldCorpus(object):
n += len(sent_tuples[1])
if self.limit and i >= self.limit:
break
i += len(paragraph_tuples)
i += 1
return n
def train_docs(self, nlp, gold_preproc=False, max_length=None,