"examples/run_bert_classifier.py" did not exist on "3bf3f9596fe58b244ceed9e4a8d454e9717b8ea2"
Commit 97c30b73 authored by erenup's avatar erenup
Browse files

add test related code

parent d5e60e5b
......@@ -256,7 +256,7 @@ def evaluate(args, model, tokenizer, prefix="", test=False):
result = {"eval_acc": acc, "eval_loss": eval_loss}
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:
logger.info("***** Eval results {} *****".format(str(prefix) + " is test:" + str(test)))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment