RetrieverΒΆ
Retriever Classes
- BaseRetriever
- RandomRetriever
RandomRetrieverRandomRetriever.dataset_readerRandomRetriever.ice_separatorRandomRetriever.ice_eos_tokenRandomRetriever.prompt_eos_tokenRandomRetriever.ice_numRandomRetriever.index_splitRandomRetriever.test_splitRandomRetriever.index_dsRandomRetriever.test_dsRandomRetriever.acceleratorRandomRetriever.seedRandomRetriever.retrieve()
- BM25Retriever
BM25RetrieverBM25Retriever.dataset_readerBM25Retriever.ice_separatorBM25Retriever.ice_eos_tokenBM25Retriever.prompt_eos_tokenBM25Retriever.ice_numBM25Retriever.index_splitBM25Retriever.test_splitBM25Retriever.index_dsBM25Retriever.test_dsBM25Retriever.acceleratorBM25Retriever.index_corpusBM25Retriever.test_corpusBM25Retriever.bm25BM25Retriever.retrieve()
- TopkRetriever
TopkRetrieverTopkRetriever.dataset_readerTopkRetriever.ice_separatorTopkRetriever.ice_eos_tokenTopkRetriever.prompt_eos_tokenTopkRetriever.ice_numTopkRetriever.index_splitTopkRetriever.test_splitTopkRetriever.index_dsTopkRetriever.test_dsTopkRetriever.acceleratorTopkRetriever.batch_sizeTopkRetriever.modelTopkRetriever.tokenizerTopkRetriever.indexTopkRetriever.retrieve()
- VotekRetriever
VotekRetrieverVotekRetriever.dataset_readerVotekRetriever.ice_separatorVotekRetriever.ice_eos_tokenVotekRetriever.prompt_eos_tokenVotekRetriever.ice_numVotekRetriever.index_splitVotekRetriever.test_splitVotekRetriever.index_dsVotekRetriever.test_dsVotekRetriever.acceleratorVotekRetriever.batch_sizeVotekRetriever.modelVotekRetriever.tokenizerVotekRetriever.indexVotekRetriever.votek_kVotekRetriever.retrieve()
- DPPRetriever
DPPRetrieverDPPRetriever.dataset_readerDPPRetriever.ice_separatorDPPRetriever.ice_eos_tokenDPPRetriever.prompt_eos_tokenDPPRetriever.ice_numDPPRetriever.index_splitDPPRetriever.test_splitDPPRetriever.index_dsDPPRetriever.test_dsDPPRetriever.acceleratorDPPRetriever.batch_sizeDPPRetriever.modelDPPRetriever.tokenizerDPPRetriever.indexDPPRetriever.seedDPPRetriever.scale_factorDPPRetriever.retrieve()
- MDLRetriever
MDLRetrieverMDLRetriever.dataset_readerMDLRetriever.ice_separatorMDLRetriever.ice_eos_tokenMDLRetriever.prompt_eos_tokenMDLRetriever.ice_numMDLRetriever.candidate_numMDLRetriever.index_splitMDLRetriever.test_splitMDLRetriever.index_dsMDLRetriever.test_dsMDLRetriever.acceleratorMDLRetriever.batch_sizeMDLRetriever.modelMDLRetriever.tokenizerMDLRetriever.indexMDLRetriever.select_timeMDLRetriever.labelsMDLRetriever.seedMDLRetriever.retrieve()
- ZeroRetriever