explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-06-06 14:13:11 +03:00
Code Issues Packages Projects Releases Wiki Activity

Set max batch size in evaluate

Browse Source
This commit is contained in:
Matthew Honnibal 2017-06-03 13:31:33 -05:00
parent 805495af27
commit fea1144e6d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/language.py
View File

@ -301,7 +301,7 @@ class Language(object):
def evaluate(self, docs_golds): def evaluate(self, docs_golds):
docs, golds = zip(*docs_golds) docs, golds = zip(*docs_golds)
scorer = Scorer() scorer = Scorer()
for doc, gold in zip(self.pipe(docs), golds): for doc, gold in zip(self.pipe(docs, batch_size=32), golds):
scorer.score(doc, gold) scorer.score(doc, gold)
doc.tensor = None doc.tensor = None
return scorer return scorer