Fix training counter for gold preprocessing

This commit is contained in:
Matthew Honnibal 2017-06-03 14:33:39 -05:00
parent d21459f87d
commit c647a0d33e

View File

@ -194,8 +194,9 @@ class GoldCorpus(object):
def count_train(self): def count_train(self):
n = 0 n = 0
for raw_text, paragraph_tuples in self.train_tuples:
for _ in self.train_tuples: for _ in self.train_tuples:
n += 1 n += len(paragraph_tuples)
return n return n
def train_docs(self, nlp, gold_preproc=False, def train_docs(self, nlp, gold_preproc=False,