Browse Source

edit example script

tags/v0.5.0-beta
yoonlee666 5 years ago
parent
commit
ab36f3d3cd
2 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model_zoo/bert/evaluation.py
  2. +0
    -0
      tests/st/networks/models/bert/test_bert_tdt_lossscale.py

+ 1
- 1
model_zoo/bert/evaluation.py View File

@@ -141,7 +141,7 @@ def test_eval():
if cfg.task == "NER": if cfg.task == "NER":
print("Precision {:.6f} ".format(callback.TP / (callback.TP + callback.FP))) print("Precision {:.6f} ".format(callback.TP / (callback.TP + callback.FP)))
print("Recall {:.6f} ".format(callback.TP / (callback.TP + callback.FN))) print("Recall {:.6f} ".format(callback.TP / (callback.TP + callback.FN)))
print("F1 {:.6f} ".format(2*callback.TP / (2*callback.TP + callback.FP + callback.FP)))
print("F1 {:.6f} ".format(2*callback.TP / (2*callback.TP + callback.FP + callback.FN)))
else: else:
print("acc_num {} , total_num {}, accuracy {:.6f}".format(callback.acc_num, callback.total_num, print("acc_num {} , total_num {}, accuracy {:.6f}".format(callback.acc_num, callback.total_num,
callback.acc_num / callback.total_num)) callback.acc_num / callback.total_num))


tests/st/networks/models/bert/bert_tdt_lossscale.py → tests/st/networks/models/bert/test_bert_tdt_lossscale.py View File


Loading…
Cancel
Save