Merge pull request #8 from Liangtaiwan/master

fix bugs
shmsw25 · Mar 18, 2020 · 5620861 · 5620861
2 parents 83051e1 + 5f24f6a
commit 5620861
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/evaluate_qa.py b/evaluate_qa.py
@@ -187,6 +187,7 @@ def get_nbest_json(prelim_predictions):
                 nbest_json = get_nbest_json([pred for pred in prelim_predictions if \
                                              pred.paragraph_index<n])
                 all_predictions[example.qas_id].append(nbest_json[0]["text"])
+            all_predictions[example.qas_id].append(example.all_answers)
 
     if write_prediction:
         logger.info("Writing predictions to: %s" % (output_prediction_file))

diff --git a/prepro.py b/prepro.py
@@ -141,6 +141,8 @@ def is_whitespace(c):
                 start_position=start_positions_list,
                 end_position=end_positions_list,
                 switch=switches_list))
+    if "test" in input_file:
+        return examples
     n_answers = []
     for example in examples:
         has_answer=False