File tree Expand file tree Collapse file tree 2 files changed +10
-10
lines changed
applications/question_answering/supervised_qa Expand file tree Collapse file tree 2 files changed +10
-10
lines changed Original file line number Diff line number Diff 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" )
Original file line number Diff line number Diff 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" )
You can’t perform that action at this time.
0 commit comments