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

updated logging and saving metrics (#10436)

Browse Source 
* updated logging and saving metrics

* space removal
This commit is contained in:
Bhadresh Savani
2021-02-27 23:23:44 +05:30
committed by GitHub
parent f52a15897b
commit aca6288ff4
12 changed files with 70 additions and 190 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
examples/text-classification/run_tf_text_classification.py
View File

@@ -293,18 +293,10 @@ def main():
results = {}
if training_args.do_eval:
logger.info("*** Evaluate ***")
result = trainer.evaluate()
output_eval_file = os.path.join(training_args.output_dir, "eval_results.txt")
with open(output_eval_file, "w") as writer:
logger.info("***** Eval results *****")
for key, value in result.items():
logger.info(" %s = %s", key, value)
writer.write("%s = %s\n" % (key, value))
results.update(result)
trainer.log_metrics("eval", result)
trainer.save_metrics("eval", result)
results.update(result)
return results