def results_cognos(self, outfolder, queries_data, userstudy_results, num_results=10): dup_count = 0 fnames = [] for c in queries_data: tname = "".join(c["q"].split(" OR ")) lname = "".join(c["l"].split()) fname = outfolder+tname+"_"+lname if os.path.exists(fname): dup_count += 1 fname = fname+"_"+str(dup_count) fname = fname+"_nl.csv" print fname result = search_cognos("+".join(c["q"].split(" OR ")) + \ "+" + "+".join(c["l"].split())) f = open(fname, "w") Groundtruth.write_cognos_result_to_file(result, f) f.close() fnames.append(fname) time.sleep(5) add_userstudy_labels_to_result(fnames, userstudy_results)
def results_ole(self, outfolder, queries_data, userstudy_results, num_results=10, entropy=False): dup_count = 0 fnames = [] for c in queries_data: tname = "".join(c["q"].split(" OR ")) lname = "".join(c["l"].split()) #regular ole search fname = outfolder+tname+"_"+lname+"_"+str(0) if os.path.exists(fname): dup_count += 1 fname = fname+"_"+str(dup_count) fname= fname+"_nl.csv" print fname result = search_ole(" ".join(c["q"].split(" OR ")), c["l"], num_results=num_results) if entropy: result = add_entropy_to_result(result) f = open(fname, "w") Groundtruth.write_result_to_file(result, f, entropy) f.close() fnames.append(fname) time.sleep(5) add_userstudy_labels_to_result(fnames, userstudy_results)