yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-11-14 00:54:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change variable names (#1286)

Browse Source
This commit is contained in:
Branden Chan 2021-07-14 14:03:34 +02:00 committed by GitHub
parent 7717e81ecc
commit da97d81305
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tutorials/Tutorial2_Finetune_a_model_on_your_data.ipynb
View File

@ -128,9 +128,9 @@
], ],
"source": [ "source": [
"reader = FARMReader(model_name_or_path=\"distilbert-base-uncased-distilled-squad\", use_gpu=True)\n", "reader = FARMReader(model_name_or_path=\"distilbert-base-uncased-distilled-squad\", use_gpu=True)\n",
"train_data = \"data/squad20\"\n", "data_dir = \"data/squad20\"\n",
"# train_data = \"PATH/TO_YOUR/TRAIN_DATA\" \n", "# data_dir = \"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\")" "reader.train(data_dir=data_dir, train_filename=\"dev-v2.0.json\", use_gpu=True, n_epochs=1, save_dir=\"my_model\")"
] ]
}, },
{ {

6
tutorials/Tutorial2_Finetune_a_model_on_your_data.py
View File

@ -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` #**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) reader = FARMReader(model_name_or_path="distilbert-base-uncased-distilled-squad", use_gpu=True)
train_data = "data/squad20" data_dir = "data/squad20"
# train_data = "PATH/TO_YOUR/TRAIN_DATA" # data_dir = "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") 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 # 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: # However, you could also save a reader manually again via:

2
tutorials/Tutorial5_Evaluation.ipynb
View File

@ -346,7 +346,7 @@
"# Initialize Reader\n", "# Initialize Reader\n",
"from haystack.reader.farm import FARMReader\n", "from haystack.reader.farm import FARMReader\n",
"\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"
] ]
}, },
{ {

2
tutorials/Tutorial5_Evaluation.py
View File

@ -93,7 +93,7 @@ def tutorial5_evaluation():
# Initialize Reader # Initialize Reader
reader = FARMReader( reader = FARMReader(
model_name_or_path="deepset/roberta-base-squad2", model_name_or_path="deepset/roberta-base-squad2",
top_k_per_candidate=4, top_k=4,
return_no_answer=True return_no_answer=True
) )