riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

move xnli_compute_metrics to data/metrics

Browse Source
This commit is contained in:
VictorSanh 2019-11-05 12:53:08 -05:00 committed by Lysandre Debut
parent 73fe2e7385
commit bcd8dc6b48
4 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
transformers/__init__.py
View File

@ -29,7 +29,7 @@ from .data import (is_sklearn_available,
xnli_output_modes, xnli_processors, xnli_tasks_num_labels)
if is_sklearn_available():
from .data import glue_compute_metrics
from .data import glue_compute_metrics, xnli_compute_metrics
# Tokenizers
from .tokenization_utils import (PreTrainedTokenizer)

2
transformers/data/__init__.py
View File

@ -4,4 +4,4 @@ from .processors import xnli_output_modes, xnli_processors, xnli_tasks_num_label
from .metrics import is_sklearn_available
if is_sklearn_available():
from .metrics import glue_compute_metrics
from .metrics import glue_compute_metrics, xnli_compute_metrics

8
transformers/data/metrics/__init__.py
View File

@ -81,3 +81,11 @@ if _has_sklearn:
return {"acc": simple_accuracy(preds, labels)}
else:
raise KeyError(task_name)
def xnli_compute_metrics(task_name, preds, labels):
assert len(preds) == len(labels)
if task_name == "xnli":
return {"acc": simple_accuracy(preds, labels)}
else:
raise KeyError(task_name)

7
transformers/data/processors/xnli.py
View File

@ -73,13 +73,6 @@ class XnliProcessor(DataProcessor):
"""See base class."""
return ["contradiction", "entailment", "neutral"]
def xnli_compute_metrics(task_name, preds, labels):
assert len(preds) == len(labels)
if task_name == "xnli":
return {"acc": simple_accuracy(preds, labels)}
else:
raise ValueError('{} is not a supported task.'.format(task_name))
xnli_processors = {
"xnli": XnliProcessor,
}