Skip to content

Commit 04142e3

Browse files
authored
[faq] fix bug of evaluate.py (#7840)
1 parent 2a5dba9 commit 04142e3

File tree

2 files changed

+10
-10
lines changed

2 files changed

+10
-10
lines changed

applications/question_answering/supervised_qa/faq_finance/evaluate.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -59,16 +59,16 @@ def recall(rs, N=10):
5959
with open(args.recall_result_file, "r", encoding="utf-8") as f:
6060
relevance_labels = []
6161
for index, line in enumerate(f):
62-
63-
if index % args.recall_num == 0 and index != 0:
64-
rs.append(relevance_labels)
65-
relevance_labels = []
66-
6762
text, recalled_text, cosine_sim = line.rstrip().split("\t")
6863
if text2similar[text] == recalled_text:
6964
relevance_labels.append(1)
7065
else:
7166
relevance_labels.append(0)
67+
68+
if (index + 1) % args.recall_num == 0:
69+
rs.append(relevance_labels)
70+
relevance_labels = []
71+
7272
recall_N = []
7373
recall_num = [1, 5, 10]
7474
result = open("result.tsv", "a")

applications/question_answering/supervised_qa/faq_system/evaluate.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -59,16 +59,16 @@ def recall(rs, N=10):
5959
with open(args.recall_result_file, "r", encoding="utf-8") as f:
6060
relevance_labels = []
6161
for index, line in enumerate(f):
62-
63-
if index % args.recall_num == 0 and index != 0:
64-
rs.append(relevance_labels)
65-
relevance_labels = []
66-
6762
text, recalled_text, cosine_sim = line.rstrip().split("\t")
6863
if text2similar[text] == recalled_text:
6964
relevance_labels.append(1)
7065
else:
7166
relevance_labels.append(0)
67+
68+
if (index + 1) % args.recall_num == 0:
69+
rs.append(relevance_labels)
70+
relevance_labels = []
71+
7272
recall_N = []
7373
recall_num = [1, 5, 10]
7474
result = open("result.tsv", "a")

0 commit comments

Comments
 (0)