auc images

This commit is contained in:
glynpu 2023-03-17 16:04:58 +08:00
parent 1162d74dc4
commit 29760e255b
3 changed files with 5 additions and 6 deletions

View File

@ -6,5 +6,5 @@ AUC results for different epochs could be found at <https://huggingface.co/GuoLi
E.g. for epoch 15 and avg 1, result log file is: <https://huggingface.co/GuoLiyong/himia_ctc_tdnn_baseline/blob/main/exp_max_duration_100/post/epoch_15-avg_1/log/log-auc-himia_aishell-2023-03-16-17-42-14>
Corresponding ROC curve is: <https://huggingface.co/GuoLiyong/himia_ctc_tdnn_baseline/blob/main/exp_max_duration_100/post/epoch_15-avg_1/himia_aishell.pdf>
Corresponding ROC curve is: <https://huggingface.co/GuoLiyong/himia_ctc_tdnn_baseline/blob/main/exp_max_duration_100/post/epoch_15-avg_1/himia_aishell.png>

View File

@ -10,8 +10,7 @@ Here is the result for epoch_15-avg_1(with the highest AUC).
| ---- | ---- | ----|
| AUC | 0.9597 |0.9292|
HiMia-CW
![himia_cw](https://huggingface.co/GuoLiyong/himia_ctc_tdnn_baseline/resolve/main/exp_max_duration_100/post/epoch_15-avg_1/himia_cw.pdf)
![himia_aishell](https://huggingface.co/GuoLiyong/himia_ctc_tdnn_baseline/resolve/main/exp_max_duration_100/post/epoch_15-avg_1/himia_aishell.png)
![himia_cw](https://huggingface.co/GuoLiyong/himia_ctc_tdnn_baseline/resolve/main/exp_max_duration_100/post/epoch_15-avg_1/himia_cw.png)
HiMia-Aishell
![himia_aishell](https://huggingface.co/GuoLiyong/himia_ctc_tdnn_baseline/resolve/main/exp_max_duration_100/post/epoch_15-avg_1/himia_aishell.pdf)

View File

@ -124,7 +124,7 @@ def main():
output_path = Path(args.positive_score_file).parent
logging.info(f"AUC of {args.legend} {output_path}: {roc_auc}")
plt.savefig(f"{output_path}/{args.legend}.pdf", bbox_inches="tight")
plt.savefig(f"{output_path}/{args.legend}.png", bbox_inches="tight")
if __name__ == "__main__":