File size: 344 Bytes
aabdf81
 
8314602
aabdf81
 
 
 
 
a066522
 
1
2
3
4
5
6
7
8
9
10

INDEX_DIR = 'data/index'
QUESTIONS_PATH = 'data/questions/selected_questions.txt'
RETRIEVER_MODEL = "sentence-transformers/multi-qa-mpnet-base-dot-v1"
RETRIEVER_MODEL_FORMAT = "sentence_transformers"
READER_MODEL = "deepset/roberta-base-squad2"
READER_CONFIG_THRESHOLD = 0.15
RETRIEVER_TOP_K = 10
READER_TOP_K = 5
LOW_RELEVANCE_THRESHOLD = 0.5