[neural search][fix bug of evaluate.py] (#7832)

* [fix][fix bug of evaluate.py]

* Update evaluate.py

* Update evaluate.py

Author: ZeyuTeng96

applications/neural_search/recall/in_batch_negative/evaluate.py

@@ -13,18 +13,29 @@
 # limitations under the License.
 
 import argparse
+import time
 
 import numpy as np
 
-import time
-
 parser = argparse.ArgumentParser()
-parser.add_argument("--similar_text_pair", type=str,
-                    default='', help="The full path of similar pair file")
-parser.add_argument("--recall_result_file", type=str,
-                    default='', help="The full path of recall result file")
-parser.add_argument("--recall_num", type=int, default=10,
-                    help="Most similar number of doc recalled from corpus per query")
+parser.add_argument(
+    "--similar_text_pair",
+    type=str,
+    default="",
+    help="The full path of similar pair file",
+)
+parser.add_argument(
+    "--recall_result_file",
+    type=str,
+    default="",
+    help="The full path of recall result file",
+)
+parser.add_argument(
+    "--recall_num",
+    type=int,
+    default=10,
+    help="Most similar number of doc recalled from corpus per query",
+)
 
 
 args = parser.parse_args()
@@ -62,7 +73,6 @@ def recall(rs, N=10):
     with open(args.recall_result_file, "r", encoding="utf-8") as f:
         relevance_labels = []
         for index, line in enumerate(f):
-
             if index % args.recall_num == 0 and index != 0:
                 rs.append(relevance_labels)
                 relevance_labels = []
@@ -73,6 +83,8 @@ def recall(rs, N=10):
             else:
                 relevance_labels.append(0)
 
+        rs.append(relevance_labels)
+
     recall_N = []
     recall_num = [1, 5, 10, 20, 50]
     for topN in recall_num: