Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
Menu
Open sidebar
nlp
ahisto-modules
Named Entity Recognition Experiments
Commits
b36d1d79
Commit
b36d1d79
authored
Sep 19, 2022
by
Vít Novotný
Browse files
Rename `AggregateMeanFScore` to `AggregateMeanFScoreEvaluator`
parent
6c760a64
Pipeline
#147289
passed with stage
in 9 minutes
Changes
3
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
ahisto_named_entity_search/recognition/__init__.py
View file @
b36d1d79
from
.evaluator
import
(
AggregateMeanFScore
,
AggregateMeanFScore
Evaluator
,
)
from
.model
import
(
...
...
@@ -14,7 +14,7 @@ from .schedule import (
__all__
=
[
'AggregateMeanFScore'
,
'AggregateMeanFScore
Evaluator
'
,
'get_schedule'
,
'NerModel'
,
'ScheduleName'
,
...
...
ahisto_named_entity_search/recognition/evaluator.py
View file @
b36d1d79
...
...
@@ -18,7 +18,7 @@ GroupMap = Dict[GroupName, Group]
FScore
=
float
class
AggregateMeanFScore
(
TokenClassificationEvaluator
):
class
AggregateMeanFScore
Evaluator
(
TokenClassificationEvaluator
):
GROUPS
:
GroupMap
=
{
'PER'
:
{
'B-PER'
,
'I-PER'
,
'B-ORG'
,
'I-ORG'
},
'LOC'
:
{
'B-LOC'
,
'I-LOC'
,
'B-ORG'
,
'I-ORG'
},
...
...
ahisto_named_entity_search/recognition/model.py
View file @
b36d1d79
...
...
@@ -16,13 +16,13 @@ from ..config import CONFIG as _CONFIG
from
..document
import
Document
,
Sentence
from
..search
import
TaggedSentence
,
NerTags
from
.schedule
import
ScheduleName
,
get_schedule
from
.evaluator
import
AggregateMeanFScore
,
FScore
,
CategoryMap
,
CategoryName
from
.evaluator
import
AggregateMeanFScore
Evaluator
,
FScore
,
CategoryMap
,
CategoryName
LOGGER
=
getLogger
(
__name__
)
EvaluationResult
=
Dict
[
AggregateMeanFScore
,
FScore
]
EvaluationResult
=
Dict
[
AggregateMeanFScore
Evaluator
,
FScore
]
class
NerModel
:
...
...
@@ -172,12 +172,12 @@ def load_ner_dataset(tagged_sentence_basename: str) -> Tuple[List[Sentence], Lis
return
ner_texts
,
all_ner_tags
def
get_evaluators
(
labels
:
Iterable
[
str
])
->
Iterable
[
AggregateMeanFScore
]:
def
get_evaluators
(
labels
:
Iterable
[
str
])
->
Iterable
[
AggregateMeanFScore
Evaluator
]:
category_map
:
CategoryMap
=
{
category
:
category_index
for
category_index
,
category
in
enumerate
(
sorted
(
labels
))
}
for
group_name
in
AggregateMeanFScore
.
get_all_group_names
():
yield
AggregateMeanFScore
(
category_map
,
group_name
,
decides_convergence
=
False
)
yield
AggregateMeanFScore
(
category_map
,
None
,
decides_convergence
=
True
)
for
group_name
in
AggregateMeanFScore
Evaluator
.
get_all_group_names
():
yield
AggregateMeanFScore
Evaluator
(
category_map
,
group_name
,
decides_convergence
=
False
)
yield
AggregateMeanFScore
Evaluator
(
category_map
,
None
,
decides_convergence
=
True
)
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment