예제 #1
0
# pylint: disable=C0413
from onir import util
registry = util.Registry(default='trivial')
register = registry.register

from onir.rankers.base import Ranker
from onir.rankers import base, conv_knrm, drmm, duetl, knrm, matchpyramid, pacrr, trivial
from onir.rankers import vanilla_transformer, cedr_drmm, cedr_knrm, cedr_pacrr, epic
from onir.rankers import matchzoo
예제 #2
0
# pylint: disable=C0413
from onir import util

registry = util.Registry(default='reranker')
register = registry.register

from onir.predictors.base import BasePredictor
from onir.predictors.reranker import Reranker
예제 #3
0
# pylint: disable=C0413
from onir import util
registry = util.Registry(default='flex')
register = registry.register

from onir.datasets.base import Dataset
from onir.datasets.index_backed import IndexBackedDataset, LazyDataRecord
from onir.datasets.multilingual_dataset import MultilingualDataset
from onir.datasets import antique, base, car, index_backed, msmarco, random, robust, flex, wikir, nyt, covid
from onir.datasets import trec_arabic, trec_mandarin, trec_spanish

# Default iteration functions over datasets
from onir.datasets.query_iter import QueryIter as query_iter
from onir.datasets.doc_iter import DocIter as doc_iter
from onir.datasets.pair_iter import pair_iter
from onir.datasets.record_iter import record_iter, run_iter, qrels_iter, pos_qrels_iter
예제 #4
0
# pylint: disable=C0413
from onir import util
registry = util.Registry(
    default='pairwise_ewc')  # Change this line to change trainer!
register = registry.register

from onir.trainers import misc
from onir.trainers.base import Trainer
from onir.trainers.trivial import TrivialTrainer
from onir.trainers.pointwise import PointwiseTrainer
from onir.trainers.pairwise import PairwiseTrainer
from onir.trainers.pairwise_ewc import PairwiseEWCTrainer
예제 #5
0
# pylint: disable=C0413
from onir import util
registry = util.Registry(default='default')
register = registry.register

from onir.pipelines.base import BasePipeline
from onir.pipelines.default import DefaultPipeline
from onir.pipelines.grid_search import GridSearchPipeline
from onir.pipelines.extract_bert_weights import ExtractBertWeights
from onir.pipelines.epic_vectorize import EpicVectorize
from onir.pipelines.tune_rerank_threshold import TuneRerankThreshold
from onir.pipelines.epic_predict import EpicPredictionPipeline
예제 #6
0
# pylint: disable=C0413
from onir import util
registry = util.Registry(default='pairwise')
register = registry.register

from onir.trainers import misc
from onir.trainers.base import Trainer
from onir.trainers.trivial import TrivialTrainer
from onir.trainers.pointwise import PointwiseTrainer
from onir.trainers.pairwise import PairwiseTrainer

from onir.trainers.pairwise_cl import PairwiseClTrainer
from onir.trainers.pointwise_cl import PointwiseClTrainer
예제 #7
0
# pylint: disable=C0413
from onir import util

registry = util.Registry(default='wordvec_hash')
register = registry.register

from onir.vocab.base import Vocab, VocabEncoder
from onir.vocab.trivial_vocab import TrivialVocab
from onir.vocab.bert_vocab import BertVocab
from onir.vocab.wordvec_vocab import WordvecVocab, WordvecUnkVocab, WordvecHashVocab