riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

no numpy

Browse Source
This commit is contained in:
thomwolf 2019-06-18 17:00:52 +02:00
parent 4e6edc3274
commit 92e0ad5aba
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/run_classifier.py
View File

@ -456,7 +456,7 @@ def main():
preds = np.argmax(preds, axis=1)
elif output_mode == "regression":
preds = np.squeeze(preds)
result = compute_metrics(task_name, preds, out_label_ids.numpy())
result = compute_metrics(task_name, preds, out_label_ids)
if args.local_rank != -1:
# Average over distributed nodes if needed
@ -533,7 +533,7 @@ def main():
eval_loss = eval_loss / nb_eval_steps
preds = preds[0]
preds = np.argmax(preds, axis=1)
result = compute_metrics(task_name, preds, out_label_ids.numpy())
result = compute_metrics(task_name, preds, out_label_ids)
if args.local_rank != -1:
# Average over distributed nodes if needed