Add comment to use documents at top 5

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:')