AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

format

Browse Source
This commit is contained in:
mlxu995 2021-12-16 12:50:10 +08:00
parent 4dd49e0f5c
commit 8383ae2f90
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
kws/bin/plot_det_curve.py
View File

@ -45,9 +45,10 @@ def plot_det_curve(keywords, stats_dir, figure_file):
plt.xlim([0, 5])
plt.ylim([0, 0.35])
plt.xticks([0, 1, 2, 3, 4, 5], ['0', '1', '2', '3', '4', '5'])
plt.yticks([0, 0.05, 0.10, 0.15, 0.20, 0.25, 0.30, 0.35], ['0', '5','10','15','20','25','30','35'])
plt.yticks([0, 0.05, 0.10, 0.15, 0.20, 0.25, 0.30, 0.35],
['0', '5', '10', '15', '20', '25', '30', '35'])
plt.xlabel('False Alarm Per Hour')
plt.ylabel('False Rejection Rate (\%)')
plt.ylabel('False Rejection Rate (\\%)')
plt.grid(linestyle='--')
plt.legend(loc='best', fontsize=16)
plt.savefig(figure_file)
@ -55,7 +56,8 @@ def plot_det_curve(keywords, stats_dir, figure_file):
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='plot det curve')
parser.add_argument('--keywords', required=True, help='keywords, must in the same order as in "dict/words.txt" separated by ", "')
parser.add_argument('--keywords', required=True,
help='keywords, must in the same order as in "dict/words.txt" separated by ", "')
parser.add_argument('--stats_dir', required=True, help='dir of stats files')
parser.add_argument('--figure_file', required=True, help='path to save det curve')