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