explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-24 17:06:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

* Print parse if verbose in scorer

Browse Source
This commit is contained in:
Matthew Honnibal 2015-04-05 22:29:30 +02:00
parent f26f381b0e
commit 021c972137
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/scorer.py
View File

@ -1,5 +1,6 @@
from __future__ import division from __future__ import division
class Scorer(object): class Scorer(object):
def __init__(self, eval_punct=False): def __init__(self, eval_punct=False):
self.heads_corr = 0 self.heads_corr = 0
@ -45,6 +46,8 @@ class Scorer(object):
self.mistokened += 1 self.mistokened += 1
if not self.skip_token(i, token, gold): if not self.skip_token(i, token, gold):
self.total += 1 self.total += 1
if verbose:
print token.orth_, token.dep_, token.head.orth_
if token.head.i == gold.heads[i]: if token.head.i == gold.heads[i]:
self.heads_corr += 1 self.heads_corr += 1
self.labels_corr += token.dep_ == gold.labels[i] self.labels_corr += token.dep_ == gold.labels[i]
@ -68,4 +71,4 @@ class Scorer(object):
self.ents_fp += len(guess_ents - gold_ents) self.ents_fp += len(guess_ents - gold_ents)
def skip_token(self, i, token, gold): def skip_token(self, i, token, gold):
return gold.labels[i] == 'P' return gold.labels[i] in ('P', 'punct')