PyRetri/pyretri/index/re_ranker/__init__.py

15 lines
364 B
Python

# -*- coding: utf-8 -*-
from yacs.config import CfgNode
from .re_ranker_impl.identity import Identity
from .re_ranker_impl.k_reciprocal import KReciprocal
from .re_ranker_impl.query_expansion import QE
from .re_ranker_impl.qe_kr import QEKR
from .re_ranker_base import ReRankerBase
__all__ = [
'ReRankerBase',
'Identity', 'KReciprocal', 'QE', 'QEKR',
]