explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-10-30 23:47:31 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix GPU evaluation

Browse Source
This commit is contained in:
Matthew Honnibal 2017-05-18 08:31:15 -05:00
parent c2c825127a
commit bbb59e371c
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
spacy/train.py
View File

@ -71,7 +71,8 @@ class Trainer(object):
self.nr_epoch += 1
def evaluate(self, dev_sents, gold_preproc=False):
scorer = Scorer()
all_docs = []
all_golds = []
for raw_text, paragraph_tuples in dev_sents:
if gold_preproc:
raw_text = None
@ -79,11 +80,10 @@ class Trainer(object):
paragraph_tuples = merge_sents(paragraph_tuples)
docs = self.make_docs(raw_text, paragraph_tuples)
golds = self.make_golds(docs, paragraph_tuples)
for doc, gold in zip(docs, golds):
state = {}
for process in self.nlp.pipeline:
assert state is not None, process.name
state = process(doc, state=state)
all_docs.extend(docs)
all_golds.extend(golds)
scorer = Scorer()
for doc, gold in zip(self.nlp.pipe(all_docs), all_golds):
scorer.score(doc, gold)
return scorer