in results/summarize_presupposition_results.py [0:0]
def unify_results(results_file, data_file):
results = np.load(os.path.join(presuppositions_dir, results_file))
data = [l for l in jsonlines.Reader(open(os.path.join(data_dir, data_file))).iter()]
for r, d in zip(results, data):
d["pred_entailment"] = r[0]
d["pred_neutral"] = r[1]
d["pred_contradiction"] = r[2]
return data