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

add test related code

Browse Source
This commit is contained in:
erenup
2019-08-20 16:31:04 +08:00
parent d5e60e5b7a
commit 97c30b73d5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/single_model_scripts/run_multiple_choice.py
View File

@@ -256,7 +256,7 @@ def evaluate(args, model, tokenizer, prefix="", test=False):
result = {"eval_acc": acc, "eval_loss": eval_loss} result = {"eval_acc": acc, "eval_loss": eval_loss}
results.update(result) results.update(result)
output_eval_file = os.path.join(eval_output_dir, "eval_results.txt") output_eval_file = os.path.join(eval_output_dir, "is_test_" + str(test) + "_eval_results.txt")
with open(output_eval_file, "w") as writer: with open(output_eval_file, "w") as writer:
logger.info("***** Eval results {} *****".format(str(prefix) + " is test:" + str(test))) logger.info("***** Eval results {} *****".format(str(prefix) + " is test:" + str(test)))