steffen/masks_augments_compare-21
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

InterSpeech Camera Ready Reporting

Browse Source
This commit is contained in:
Si11ium 2020-08-06 08:12:08 +02:00
parent 256e09a8b9
commit b57eabb371
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
util/metric_reader.py
View File

@ -37,7 +37,7 @@ if __name__ == '__main__':
metric_dict[header].append(float(value))
except ValueError:
metric_dict[header].append(value)
for score, func in zip(['mean', 'max', 'median', 'std'], [np.mean, np.max, np.median, np.std]):
for score, func in zip(['max'], [np.max]): # [np.mean, np.max, np.median, np.std]):
try:
uar_scores[score].append(round(func(np.asarray(metric_dict['uar_score'])) * 100, 2))
except ValueError as e:
@ -51,7 +51,7 @@ if __name__ == '__main__':
if not file_existed:
writer.writeheader() # file doesn't exist yet, write a header
try:
for row_idx in range(len(uar_scores['mean'])):
for row_idx in range(len(uar_scores['max'])):
writer.writerow({key: uar_scores[key][row_idx] for key in headers})
except IndexError:
print('could not read: ')