Spaces:
Running
Running
Update Space (evaluate main: 5aa3982a)
Browse files- f1.py +1 -1
- requirements.txt +1 -1
f1.py
CHANGED
|
@@ -127,4 +127,4 @@ class F1(evaluate.Metric):
|
|
| 127 |
score = f1_score(
|
| 128 |
references, predictions, labels=labels, pos_label=pos_label, average=average, sample_weight=sample_weight
|
| 129 |
)
|
| 130 |
-
return {"f1":
|
|
|
|
| 127 |
score = f1_score(
|
| 128 |
references, predictions, labels=labels, pos_label=pos_label, average=average, sample_weight=sample_weight
|
| 129 |
)
|
| 130 |
+
return {"f1": score if getattr(score, "size", 1) > 1 else float(score)}
|
requirements.txt
CHANGED
|
@@ -1,2 +1,2 @@
|
|
| 1 |
-
git+https://github.com/huggingface/evaluate@
|
| 2 |
scikit-learn
|
|
|
|
| 1 |
+
git+https://github.com/huggingface/evaluate@5aa3982a9a8c86e506860e381d428a64b0cce73b
|
| 2 |
scikit-learn
|