Для этого я использую следующий код: < /p>
Код: Выделить всё
training_args = TrainingArguments(
output_dir=resume_from_checkpoint,
evaluation_strategy="epoch",
per_device_train_batch_size=1,
)
def compute_metrics(pred: EvalPrediction):
labels = pred.label_ids
preds = pred.predictions.argmax(-1)
f1 = f1_score(labels, preds, average="weighted")
acc = accuracy_score(labels, preds, average="weighted")
return {"accuracy": acc, "f1": f1}
trainer = Trainer(
model=self.nli_model,
args=training_args,
train_dataset=tokenized_datasets,
compute_metrics=compute_metrics,
)
< /code>
Однако я получаю < /p>
ValueError: Trainer: evaluation requires an eval_dataset
Подробнее здесь: https://stackoverflow.com/questions/763 ... ce-transfo