Spaces:
Build error
Build error
File size: 934 Bytes
e62781a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
#!/usr/bin/env python3
# Copyright 2017-present, Facebook, Inc.
# All rights reserved.
#
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.
import os
from .. import DATA_DIR
DEFAULTS = {
'db_path': os.path.join(DATA_DIR, 'wikipedia/docs.db'),
'tfidf_path': os.path.join(
DATA_DIR,
'wikipedia/docs-tfidf-ngram=2-hash=16777216-tokenizer=simple.npz'
),
'elastic_url': 'localhost:9200'
}
def set_default(key, value):
global DEFAULTS
DEFAULTS[key] = value
def get_class(name):
if name == 'tfidf':
return TfidfDocRanker
if name == 'sqlite':
return DocDB
if name == 'elasticsearch':
return ElasticDocRanker
raise RuntimeError('Invalid retriever class: %s' % name)
from .doc_db import DocDB
from .tfidf_doc_ranker import TfidfDocRanker
from .elastic_doc_ranker import ElasticDocRanker
|