Skip to content

Commit ff3cf2e

Browse files
committed
try to fix "eval_f1" not found error
1 parent 8d9255e commit ff3cf2e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

autointent/modules/scoring/_bert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ def tokenize_function(examples: dict[str, Any]) -> dict[str, Any]:
132132

133133
tokenized_dataset = dataset.map(tokenize_function, batched=True, batch_size=self.batch_size)
134134

135-
metric_name = "f1"
135+
metric_name = "eval_f1"
136136

137137
def compute_metrics(predictions: EvalPrediction) -> dict[str, float]:
138138
return {metric_name: scoring_f1(predictions.label_ids.tolist(), predictions.predictions.tolist())} # type: ignore[union-attr]

0 commit comments

Comments
 (0)