Commit 7e5701eb authored by Vít Novotný's avatar Vít Novotný
Browse files

Add `*-ORG` and `*-MISC` to `AggregateMeanFScore.GROUPS`

parent 63feb75b
......@@ -20,9 +20,9 @@ FScore = float
class AggregateMeanFScore(TokenClassificationEvaluator):
GROUPS: GroupMap = {
'PER': ('B-PER', 'I-PER'),
'LOC': ('B-LOC', 'I-LOC'),
'O': ('O', ),
'PER': ('B-PER', 'I-PER', 'B-ORG', 'I-ORG'),
'LOC': ('B-LOC', 'I-LOC', 'B-ORG', 'I-ORG'),
'O': ('O', 'B-MISC', 'I-MISC'),
def __init__(self, category_map: CategoryMap, group_name: Optional[GroupName],
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment