Spaces:
Running
Running
Update Space (evaluate main: a45df1eb)
Browse files
glue.py
CHANGED
|
@@ -103,7 +103,7 @@ def pearson_and_spearman(preds, labels):
|
|
| 103 |
|
| 104 |
|
| 105 |
@evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
|
| 106 |
-
class Glue(evaluate.
|
| 107 |
def _info(self):
|
| 108 |
if self.config_name not in [
|
| 109 |
"sst2",
|
|
@@ -124,7 +124,7 @@ class Glue(evaluate.EvaluationModule):
|
|
| 124 |
'["sst2", "mnli", "mnli_mismatched", "mnli_matched", '
|
| 125 |
'"cola", "stsb", "mrpc", "qqp", "qnli", "rte", "wnli", "hans"]'
|
| 126 |
)
|
| 127 |
-
return evaluate.
|
| 128 |
description=_DESCRIPTION,
|
| 129 |
citation=_CITATION,
|
| 130 |
inputs_description=_KWARGS_DESCRIPTION,
|
|
|
|
| 103 |
|
| 104 |
|
| 105 |
@evaluate.utils.file_utils.add_start_docstrings(_DESCRIPTION, _KWARGS_DESCRIPTION)
|
| 106 |
+
class Glue(evaluate.Metric):
|
| 107 |
def _info(self):
|
| 108 |
if self.config_name not in [
|
| 109 |
"sst2",
|
|
|
|
| 124 |
'["sst2", "mnli", "mnli_mismatched", "mnli_matched", '
|
| 125 |
'"cola", "stsb", "mrpc", "qqp", "qnli", "rte", "wnli", "hans"]'
|
| 126 |
)
|
| 127 |
+
return evaluate.MetricInfo(
|
| 128 |
description=_DESCRIPTION,
|
| 129 |
citation=_CITATION,
|
| 130 |
inputs_description=_KWARGS_DESCRIPTION,
|