Skip to content

Commit 312c611

Browse files
committed
fix: typo
1 parent e1e6ca1 commit 312c611

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

anyclassifier/fasttext_wrapper/trainer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -180,9 +180,9 @@ def evaluate(self, dataset: Dataset) -> Dict[str, float]:
180180
metric_kwargs = self.metric_kwargs or {}
181181
if isinstance(self.metric, str):
182182
metric_fn = evaluate.load(self.metric)
183-
results = metric_fn.compute(predictions=y_pred, references=y_test, **metric_kwargs)
183+
results = metric_fn.compute(predictions=label_pred, references=label, **metric_kwargs)
184184
elif callable(self.metric):
185-
results = self.metric(y_pred, y_test, **metric_kwargs)
185+
results = self.metric(label_pred, label, **metric_kwargs)
186186
else:
187187
raise ValueError("metric must be a string or a callable")
188188
return {"metric": results}

0 commit comments

Comments
 (0)