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
|