diff --git a/rest_api/config.py b/rest_api/config.py index 3420f7e5a..a53828703 100644 --- a/rest_api/config.py +++ b/rest_api/config.py @@ -35,7 +35,7 @@ READER_TYPE = os.getenv("READER_TYPE", "FARMReader") # alternative: 'Transformer READER_TOKENIZER = os.getenv("READER_TOKENIZER", None) CONTEXT_WINDOW_SIZE = int(os.getenv("CONTEXT_WINDOW_SIZE", 500)) DEFAULT_TOP_K_READER = int(os.getenv("DEFAULT_TOP_K_READER", 5)) # How many answers to return in total -TOP_K_PER_CANDIDATE = int(os.getenv("TOP_K_PER_CANDIDATE", 10)) # How many answers can come from one indexed doc +TOP_K_PER_CANDIDATE = int(os.getenv("TOP_K_PER_CANDIDATE", 3)) # How many answers can come from one indexed doc TOP_K_PER_SAMPLE = int(os.getenv("TOP_K_PER_SAMPLE", 1)) # How many answers can come from one passage that the reader processes at once (i.e. text of max_seq_len from the doc) NO_ANS_BOOST = int(os.getenv("NO_ANS_BOOST", -10)) READER_CAN_HAVE_NO_ANSWER = os.getenv("READER_CAN_HAVE_NO_ANSWER", "True").lower() == "true"