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

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

parent 63feb75b

ahisto_named_entity_search/recognition/evaluator.py

+3 −3
Original line number Diff line number Diff line
@@ -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],