update accuracy report column order
Browse files- lib/report.py +6 -5
lib/report.py
CHANGED
|
@@ -243,9 +243,9 @@ class AccuracyItem:
|
|
| 243 |
self.asr_accuracy = run_textdistance(self.audio_text, self.src_text)
|
| 244 |
self.text_compare = highlight_diff(self.audio_text, self.src_text)
|
| 245 |
def to_list(self):
|
| 246 |
-
return [self.audio, self.translation_type, self.audio_length,
|
| 247 |
-
self.
|
| 248 |
-
self.
|
| 249 |
|
| 250 |
class AccuracyReport:
|
| 251 |
items:List[AccuracyItem] = []
|
|
@@ -258,8 +258,9 @@ class AccuracyReport:
|
|
| 258 |
def to_csv(self, csv_path):
|
| 259 |
print("accuracy item length: ", len(self.items))
|
| 260 |
self.print_summary()
|
| 261 |
-
header = ["audio_name", "translation", "audio_length",
|
| 262 |
-
"distance", "normalized distance",
|
|
|
|
| 263 |
save_csv(csv_path, header, [i.to_list() for i in self.items])
|
| 264 |
|
| 265 |
|
|
|
|
| 243 |
self.asr_accuracy = run_textdistance(self.audio_text, self.src_text)
|
| 244 |
self.text_compare = highlight_diff(self.audio_text, self.src_text)
|
| 245 |
def to_list(self):
|
| 246 |
+
return [self.audio, self.translation_type, self.audio_length,
|
| 247 |
+
self.asr_accuracy[0], self.asr_accuracy[1],
|
| 248 |
+
self.src_text, self.audio_text, self.text_compare]
|
| 249 |
|
| 250 |
class AccuracyReport:
|
| 251 |
items:List[AccuracyItem] = []
|
|
|
|
| 258 |
def to_csv(self, csv_path):
|
| 259 |
print("accuracy item length: ", len(self.items))
|
| 260 |
self.print_summary()
|
| 261 |
+
header = ["audio_name", "translation", "audio_length",
|
| 262 |
+
"distance", "normalized distance",
|
| 263 |
+
"src text", "audio text", "text compare"]
|
| 264 |
save_csv(csv_path, header, [i.to_list() for i in self.items])
|
| 265 |
|
| 266 |
|