fix #74 #36 #33

cdqa-suite · Mar 11, 2019 · 826c58b · 826c58b
1 parent a35931c
commit 826c58b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cdqa/reader/bertqa_sklearn.py b/cdqa/reader/bertqa_sklearn.py
@@ -615,6 +615,7 @@ def write_predictions(all_examples, all_features, all_results, n_best_size,
 
         if not version_2_with_negative:
             all_predictions[example.qas_id] = nbest_json[0]["text"]
+            all_nbest_json[example.qas_id] = nbest_json
         else:
             # predict "" iff the null score - the score of best non-null > threshold
             score_diff = score_null - best_non_null_entry.start_logit - (