xnli - output_mode consistency

2019-10-29 12:04:20 -04:00
parent d52e98ff9a
commit c4336ecbbd
1 changed files with 2 additions and 2 deletions
--- a/examples/run_xnli.py
+++ b/examples/run_xnli.py
@@ -247,8 +247,8 @@ def evaluate(args, model, tokenizer, prefix=""):
        eval_loss = eval_loss / nb_eval_steps
        if args.output_mode == "classification":
            preds = np.argmax(preds, axis=1)
-        elif args.output_mode == "regression":
-            preds = np.squeeze(preds)
+        else:
+            raise ValueError(f'No other `output_mode` for XNLI.')
        result = compute_metrics(eval_task, preds, out_label_ids)
        results.update(result)