From 9601a558f83de6f86af5ad5dbbaa70a9f12ff794 Mon Sep 17 00:00:00 2001
From: feger <marc.feger@hhu.de>
Date: Wed, 17 Apr 2019 20:13:27 +0200
Subject: [PATCH] Add comment to use documents at top 5
---
Aufgabe2/rocchio.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/Aufgabe2/rocchio.py b/Aufgabe2/rocchio.py
index 7f8e7e9..583f694 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:')
--
GitLab