Stefano Fiorucci
refactoring
a066522
raw
history blame
344 Bytes
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