Fix Language.evaluate

2017-08-06 02:18:31 +02:00 · 2017-08-06 02:18:31 +02:00 · 0eec7c9e9b
parent 0a566dc320
commit 0eec7c9e9b
1 changed files with 6 additions and 2 deletions
--- a/spacy/language.py
+++ b/spacy/language.py
@ -382,9 +382,13 @@ class Language(object):
        return optimizer

    def evaluate(self, docs_golds):
-        docs, golds = zip(*docs_golds)
        scorer = Scorer()
-        for doc, gold in zip(self.pipe(docs, batch_size=32), golds):
+        docs, golds = zip(*docs_golds)
+        docs = list(docs)
+        golds = list(golds)
+        for pipe in self.pipeline:
+            docs = pipe.pipe(docs)
+        for doc, gold in zip(docs, golds):
            scorer.score(doc, gold)
            doc.tensor = None
        return scorer