From 1566da184978412d6bfa7c10983fdb0485369b31 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gabriela=20Pa=C5=82ka?= <gabi179@gmail.com>
Date: Fri, 3 Apr 2020 16:28:38 +0200
Subject: [PATCH] Add sum for wer

---
 count_metrics.py | 4 ++--
 script.sh        | 2 ++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/count_metrics.py b/count_metrics.py
index 8ede62e..b386e63 100755
--- a/count_metrics.py
+++ b/count_metrics.py
@@ -6,8 +6,8 @@ if __name__ == '__main__':
     correct_sentences = 0
 
     with open(sys.argv[1], 'r') as sclite_result, \
-            open('wer.txt', 'a+') as wer, \
-            open('srr.txt', 'a+') as srr:
+            open('wer.txt', 'w') as wer, \
+            open('srr.txt', 'w') as srr:
 
         for line in sclite_result:
             complited = False
diff --git a/script.sh b/script.sh
index 2c0e2ad..87814c1 100755
--- a/script.sh
+++ b/script.sh
@@ -6,3 +6,5 @@ cut -f3 $1 | awk 'BEGIN{FS=OFS="\t"}{print $0,"(sp1_"NR")"}' > reference.trn
 sclite -f 0 -r reference.trn trn -h hypothesis.trn trn -e utf-8 -i rm -o all stdout > results.txt
 python3 count_metrics.py results.txt
 paste results.txt wer.txt > tmp.txt && mv tmp.txt results.txt
+cut -f5 results.txt
+awk '{ total += $1; count++ } END { print total/count }' wer.txt > average && mv average wer.txt
\ No newline at end of file