Skip to content
This repository was archived by the owner on Nov 8, 2022. It is now read-only.

Commit b9c9436

Browse files
authored
Merge pull request #1 from NervanaSystems/peter/conlleval_fix
fix conlleval script
2 parents d08f349 + 5a81673 commit b9c9436

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

nlp_architect/utils/conlleval.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,22 @@ def __init__(self):
5959
self.t_found_correct = defaultdict(int)
6060
self.t_found_guessed = defaultdict(int)
6161

62+
def parse_args(argv):
63+
import argparse
64+
parser = argparse.ArgumentParser(
65+
description='evaluate tagging results using CoNLL criteria',
66+
formatter_class=argparse.ArgumentDefaultsHelpFormatter
67+
)
68+
arg = parser.add_argument
69+
arg('-b', '--boundary', metavar='STR', default='-X-',
70+
help='sentence boundary')
71+
arg('-d', '--delimiter', metavar='CHAR', default=ANY_SPACE,
72+
help='character delimiting items in input')
73+
arg('-o', '--otag', metavar='CHAR', default='O',
74+
help='alternative outside tag')
75+
arg('file', nargs='?', default=None)
76+
return parser.parse_args(argv)
77+
6278
def parse_tag(t):
6379
m = re.match(r'^([^-]*)-(.*)$', t)
6480
return m.groups() if m else (t, '')

0 commit comments

Comments
 (0)