diff --git a/docs/_src/api/api/reader.md b/docs/_src/api/api/reader.md index d6d196144..48afb3c37 100644 --- a/docs/_src/api/api/reader.md +++ b/docs/_src/api/api/reader.md @@ -564,7 +564,7 @@ With this reader, you can directly get predictions via predict() #### \_\_init\_\_ ```python -def __init__(model_name_or_path: str = "distilbert-base-uncased-distilled-squad", model_version: Optional[str] = None, tokenizer: Optional[str] = None, context_window_size: int = 70, use_gpu: bool = True, top_k: int = 10, top_k_per_candidate: int = 4, return_no_answers: bool = True, max_seq_len: int = 256, doc_stride: int = 128) +def __init__(model_name_or_path: str = "distilbert-base-uncased-distilled-squad", model_version: Optional[str] = None, tokenizer: Optional[str] = None, context_window_size: int = 70, use_gpu: bool = True, top_k: int = 10, top_k_per_candidate: int = 3, return_no_answers: bool = False, max_seq_len: int = 256, doc_stride: int = 128) ``` Load a QA model from Transformers. diff --git a/haystack/json-schemas/haystack-pipeline-master.schema.json b/haystack/json-schemas/haystack-pipeline-master.schema.json index 296ea6e96..eba025abb 100644 --- a/haystack/json-schemas/haystack-pipeline-master.schema.json +++ b/haystack/json-schemas/haystack-pipeline-master.schema.json @@ -4174,12 +4174,12 @@ }, "top_k_per_candidate": { "title": "Top K Per Candidate", - "default": 4, + "default": 3, "type": "integer" }, "return_no_answers": { "title": "Return No Answers", - "default": true, + "default": false, "type": "boolean" }, "max_seq_len": { diff --git a/haystack/nodes/reader/transformers.py b/haystack/nodes/reader/transformers.py index baf51cb09..e69f4ab7e 100644 --- a/haystack/nodes/reader/transformers.py +++ b/haystack/nodes/reader/transformers.py @@ -26,8 +26,8 @@ class TransformersReader(BaseReader): context_window_size: int = 70, use_gpu: bool = True, top_k: int = 10, - top_k_per_candidate: int = 4, - return_no_answers: bool = True, + top_k_per_candidate: int = 3, + return_no_answers: bool = False, max_seq_len: int = 256, doc_stride: int = 128, ):