diff --git a/tutorials/Tutorial2_Finetune_a_model_on_your_data.ipynb b/tutorials/Tutorial2_Finetune_a_model_on_your_data.ipynb index c814f0952..e9f1a0bd3 100644 --- a/tutorials/Tutorial2_Finetune_a_model_on_your_data.ipynb +++ b/tutorials/Tutorial2_Finetune_a_model_on_your_data.ipynb @@ -128,9 +128,9 @@ ], "source": [ "reader = FARMReader(model_name_or_path=\"distilbert-base-uncased-distilled-squad\", use_gpu=True)\n", - "train_data = \"data/squad20\"\n", - "# train_data = \"PATH/TO_YOUR/TRAIN_DATA\" \n", - "reader.train(data_dir=train_data, train_filename=\"dev-v2.0.json\", use_gpu=True, n_epochs=1, save_dir=\"my_model\")" + "data_dir = \"data/squad20\"\n", + "# data_dir = \"PATH/TO_YOUR/TRAIN_DATA\" \n", + "reader.train(data_dir=data_dir, train_filename=\"dev-v2.0.json\", use_gpu=True, n_epochs=1, save_dir=\"my_model\")" ] }, { diff --git a/tutorials/Tutorial2_Finetune_a_model_on_your_data.py b/tutorials/Tutorial2_Finetune_a_model_on_your_data.py index 0af993717..4b970a269 100755 --- a/tutorials/Tutorial2_Finetune_a_model_on_your_data.py +++ b/tutorials/Tutorial2_Finetune_a_model_on_your_data.py @@ -36,9 +36,9 @@ def tutorial2_finetune_a_model_on_your_data(): #**Recommendation: Run training on a GPU. To do so change the `use_gpu` arguments below to `True` reader = FARMReader(model_name_or_path="distilbert-base-uncased-distilled-squad", use_gpu=True) - train_data = "data/squad20" - # train_data = "PATH/TO_YOUR/TRAIN_DATA" - reader.train(data_dir=train_data, train_filename="dev-v2.0.json", use_gpu=True, n_epochs=1, save_dir="my_model") + data_dir = "data/squad20" + # data_dir = "PATH/TO_YOUR/TRAIN_DATA" + reader.train(data_dir=data_dir, train_filename="dev-v2.0.json", use_gpu=True, n_epochs=1, save_dir="my_model") # Saving the model happens automatically at the end of training into the `save_dir` you specified # However, you could also save a reader manually again via: diff --git a/tutorials/Tutorial5_Evaluation.ipynb b/tutorials/Tutorial5_Evaluation.ipynb index da1cf2e34..b6402514d 100644 --- a/tutorials/Tutorial5_Evaluation.ipynb +++ b/tutorials/Tutorial5_Evaluation.ipynb @@ -346,7 +346,7 @@ "# Initialize Reader\n", "from haystack.reader.farm import FARMReader\n", "\n", - "reader = FARMReader(\"deepset/roberta-base-squad2\", top_k_per_candidate=4, return_no_answer=True)\n" + "reader = FARMReader(\"deepset/roberta-base-squad2\", top_k=4, return_no_answer=True)\n" ] }, { diff --git a/tutorials/Tutorial5_Evaluation.py b/tutorials/Tutorial5_Evaluation.py index da59b4617..1c8f2dcc6 100644 --- a/tutorials/Tutorial5_Evaluation.py +++ b/tutorials/Tutorial5_Evaluation.py @@ -93,7 +93,7 @@ def tutorial5_evaluation(): # Initialize Reader reader = FARMReader( model_name_or_path="deepset/roberta-base-squad2", - top_k_per_candidate=4, + top_k=4, return_no_answer=True )