SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

xnli - output_mode consistency

Browse Source
This commit is contained in:
VictorSanh
2019-10-29 12:04:20 -04:00
committed by Lysandre Debut
parent d52e98ff9a
commit c4336ecbbd
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/run_xnli.py
View File

@@ -247,8 +247,8 @@ def evaluate(args, model, tokenizer, prefix=""):
eval_loss = eval_loss / nb_eval_steps eval_loss = eval_loss / nb_eval_steps
if args.output_mode == "classification": if args.output_mode == "classification":
preds = np.argmax(preds, axis=1) preds = np.argmax(preds, axis=1)
elif args.output_mode == "regression": else:
preds = np.squeeze(preds) raise ValueError(f'No other `output_mode` for XNLI.')
result = compute_metrics(eval_task, preds, out_label_ids) result = compute_metrics(eval_task, preds, out_label_ids)
results.update(result) results.update(result)