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)
