Spaces:
Running
Running
Joshua Lochner
commited on
Commit
•
b3b69aa
1
Parent(s):
4f1ee08
Update evaluate.py
Browse files- src/evaluate.py +1 -2
src/evaluate.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
from datasets import load_dataset
|
2 |
from transformers import (
|
3 |
AutoModelForSeq2SeqLM,
|
@@ -41,8 +42,6 @@ class EvaluationArguments(TrainingOutputArguments):
|
|
41 |
}
|
42 |
)
|
43 |
|
44 |
-
from utils import jaccard
|
45 |
-
|
46 |
|
47 |
def attach_predictions_to_sponsor_segments(predictions, sponsor_segments):
|
48 |
"""Attach sponsor segments to closest prediction"""
|
|
|
1 |
+
from utils import jaccard
|
2 |
from datasets import load_dataset
|
3 |
from transformers import (
|
4 |
AutoModelForSeq2SeqLM,
|
|
|
42 |
}
|
43 |
)
|
44 |
|
|
|
|
|
45 |
|
46 |
def attach_predictions_to_sponsor_segments(predictions, sponsor_segments):
|
47 |
"""Attach sponsor segments to closest prediction"""
|