diff --git a/Aufgabe2/rocchio.py b/Aufgabe2/rocchio.py index 7f8e7e92356634d2cddc581783085b82ff6e3256..583f6944c7a82031a24015a82050942efd734f23 100644 --- a/Aufgabe2/rocchio.py +++ b/Aufgabe2/rocchio.py @@ -48,7 +48,7 @@ def q_m(query: Dict, documents: List[Dict], feedback: List, alpha: Double, beta: :return: None """ relevant_documents, not_relevant_documents = assemble_dr_dnr(feedback, documents) - + print(relevant_documents, not_relevant_documents) addend_1: List = list(np.multiply(alpha, dict_to_list(query.get('document_vector')))) addend_2: List = list(np.zeros(len(query.get('document_vector')))) @@ -88,6 +88,7 @@ if __name__ == '__main__': $ python rocchio.py """ all_documents = FileReader('./files/output.documents').read_document_and_eval_type_of_content() + # all_documents = sorted(all_documents, key=lambda k: k['cosine'], reverse=True)[:5] search_query = FileReader('./files/output.query').read_document_and_eval_type_of_content() print('Top-5 results:')