Commit 56b9ac15 authored by Marek Medved's avatar Marek Medved
Browse files

change position score, doc_id in document selection

parent 1e3b75a8

aqa_v2.py

+2 −2
Original line number Diff line number Diff line
@@ -16,7 +16,7 @@ def print_basic_data(data, output): 
    output.write(f"Question type: {data['q_type']}\n")

    output.write(f"Answer type: {data['a_type']}\n")

    output.write(f"Best documents: \n")

    for doc, score in data.get('ds_result', []):

    for score, doc in data.get('ds_result', []):

        output.write(f"{score:.4f}: {doc}\n")

    output.write(f"Selected answer: \n")

    for score, answer in data.get('as_result', [])[:1]:
@@ -71,7 +71,7 @@ def main(): 
            print(f'q_type: {q_type}')



        # Answer selection

        ans_sel = get_answer_selection(input_w_question, top_n_docs[0][0])

        ans_sel = get_answer_selection(input_w_question, top_n_docs[0][1])



        if args.verbose:

            print(f'ans_sel: {ans_sel}')