423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-22 13:42:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

tiny fix

Browse Source 
Former-commit-id: 73280b7dc7f8b3210bb08dfc3cf34760190f585a
This commit is contained in:
hiyouga 2024-07-01 05:43:17 +08:00
parent de4de5b5ab
commit 2cf03017a0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/train/sft/metric.py
View File

@ -48,7 +48,7 @@ def compute_accuracy(eval_preds: "EvalPrediction") -> Dict[str, float]:
preds, labels = eval_preds.predictions, eval_preds.label_ids preds, labels = eval_preds.predictions, eval_preds.label_ids
accuracies = [] accuracies = []
for i in range(len(preds)): for i in range(len(preds)):
pred, label = preds[i, 1:], labels[i, :-1] pred, label = preds[i, :-1], labels[i, 1:]
label_mask = label != IGNORE_INDEX label_mask = label != IGNORE_INDEX
accuracies.append(np.mean(pred[label_mask] == label[label_mask])) accuracies.append(np.mean(pred[label_mask] == label[label_mask]))