diff --git a/calc_wer.py b/calc_wer.py index 914d720..8e23f4a 100644 --- a/calc_wer.py +++ b/calc_wer.py @@ -31,7 +31,7 @@ for line in sys.stdin: srr = correct_sentences/len(wer_list) wer_all = wer(sum(correct), sum(substitution), sum(deletions), sum(insertions)) -data = pd.read_csv(FILE_NAME, sep='\t', header=False) +data = pd.read_csv(FILE_NAME, sep='\t', header=None) df = pd.DataFrame(data) print(len(wer_list), len(data)) df.insert(4, '', wer_list)