diff --git a/tutorials/Tutorial5_Evaluation.ipynb b/tutorials/Tutorial5_Evaluation.ipynb index 01be0ae781..646817d114 100644 --- a/tutorials/Tutorial5_Evaluation.ipynb +++ b/tutorials/Tutorial5_Evaluation.ipynb @@ -30,8 +30,7 @@ "**Runtime -> Change Runtime type -> Hardware accelerator -> GPU**\n", "\n", "" - ], - "outputs": [] + ] }, { "cell_type": "code", @@ -992,7 +991,7 @@ "outputs": [], "source": [ "advanced_eval_result = pipeline.eval(\n", - " labels=eval_labels, params={\"Retriever\": {\"top_k\": 1}}, sas_model_name_or_path=\"cross-encoder/stsb-roberta-large\"\n", + " labels=eval_labels, params={\"Retriever\": {\"top_k\": 5}}, sas_model_name_or_path=\"cross-encoder/stsb-roberta-large\"\n", ")\n", "\n", "metrics = advanced_eval_result.calculate_metrics()\n", diff --git a/tutorials/Tutorial5_Evaluation.py b/tutorials/Tutorial5_Evaluation.py index b2164c3c59..f8ef853574 100644 --- a/tutorials/Tutorial5_Evaluation.py +++ b/tutorials/Tutorial5_Evaluation.py @@ -188,7 +188,7 @@ def tutorial5_evaluation(): advanced_eval_result = pipeline.eval( labels=eval_labels, - params={"Retriever": {"top_k": 1}}, + params={"Retriever": {"top_k": 5}}, sas_model_name_or_path="cross-encoder/stsb-roberta-large", )