Spaces:
Running
Running
Joshua Lochner
commited on
Commit
·
63f1925
1
Parent(s):
69fe24d
Allow seeding for evaluations
Browse files- src/evaluate.py +5 -4
src/evaluate.py
CHANGED
|
@@ -6,8 +6,8 @@ from transformers import (
|
|
| 6 |
)
|
| 7 |
from preprocess import DatasetArguments, ProcessedArguments, get_words
|
| 8 |
from model import get_classifier_vectorizer
|
| 9 |
-
from shared import device
|
| 10 |
-
from predict import ClassifierArguments, predict,
|
| 11 |
from segment import word_start, word_end, SegmentationArguments, add_labels_to_words
|
| 12 |
import pandas as pd
|
| 13 |
from dataclasses import dataclass, field
|
|
@@ -150,10 +150,11 @@ def main():
|
|
| 150 |
EvaluationArguments,
|
| 151 |
ProcessedArguments,
|
| 152 |
SegmentationArguments,
|
| 153 |
-
ClassifierArguments
|
|
|
|
| 154 |
))
|
| 155 |
|
| 156 |
-
evaluation_args, processed_args, segmentation_args, classifier_args = hf_parser.parse_args_into_dataclasses()
|
| 157 |
|
| 158 |
model = AutoModelForSeq2SeqLM.from_pretrained(evaluation_args.model_path)
|
| 159 |
model.to(device())
|
|
|
|
| 6 |
)
|
| 7 |
from preprocess import DatasetArguments, ProcessedArguments, get_words
|
| 8 |
from model import get_classifier_vectorizer
|
| 9 |
+
from shared import device, GeneralArguments
|
| 10 |
+
from predict import ClassifierArguments, predict, add_predictions, TrainingOutputArguments
|
| 11 |
from segment import word_start, word_end, SegmentationArguments, add_labels_to_words
|
| 12 |
import pandas as pd
|
| 13 |
from dataclasses import dataclass, field
|
|
|
|
| 150 |
EvaluationArguments,
|
| 151 |
ProcessedArguments,
|
| 152 |
SegmentationArguments,
|
| 153 |
+
ClassifierArguments,
|
| 154 |
+
GeneralArguments
|
| 155 |
))
|
| 156 |
|
| 157 |
+
evaluation_args, processed_args, segmentation_args, classifier_args, _ = hf_parser.parse_args_into_dataclasses()
|
| 158 |
|
| 159 |
model = AutoModelForSeq2SeqLM.from_pretrained(evaluation_args.model_path)
|
| 160 |
model.to(device())
|