Update app.py
Browse files
app.py
CHANGED
|
@@ -169,13 +169,13 @@ def evaluate_retrieval(query: str, ground_truth_docs: List[str], retriever, k: i
|
|
| 169 |
hits = sum([1 for doc in top_k if any(gt.lower() in doc.lower() for gt in ground_truth_docs)])
|
| 170 |
precision = hits / k
|
| 171 |
recall = hits / len(ground_truth_docs) if ground_truth_docs else 0.0
|
| 172 |
-
print("\n
|
| 173 |
-
print("
|
| 174 |
for i, doc in enumerate(top_k, 1):
|
| 175 |
print(f"{i}. {doc[:200]}...")
|
| 176 |
-
print(f"\n
|
| 177 |
-
print(f"
|
| 178 |
-
print(f"
|
| 179 |
return {
|
| 180 |
f"Precision@{k}": precision,
|
| 181 |
f"Recall@{k}": recall,
|
|
|
|
| 169 |
hits = sum([1 for doc in top_k if any(gt.lower() in doc.lower() for gt in ground_truth_docs)])
|
| 170 |
precision = hits / k
|
| 171 |
recall = hits / len(ground_truth_docs) if ground_truth_docs else 0.0
|
| 172 |
+
print("\n Query:", query)
|
| 173 |
+
print(" Top-K Retrieved Documents:")
|
| 174 |
for i, doc in enumerate(top_k, 1):
|
| 175 |
print(f"{i}. {doc[:200]}...")
|
| 176 |
+
print(f"\n Evaluation Results:")
|
| 177 |
+
print(f" Precision@{k}: {precision:.2f}")
|
| 178 |
+
print(f" Recall@{k}: {recall:.2f}")
|
| 179 |
return {
|
| 180 |
f"Precision@{k}": precision,
|
| 181 |
f"Recall@{k}": recall,
|