Skip to content

Document Ranker Config

Simple Ranker

Example

gfmrag/workflow/config/doc_ranker/simple_ranker.yaml
_target_: gfmrag.doc_rankers.SimpleRanker
Parameter Options Note
_target_ gfmrag.doc_rankers.SimpleRanker The class name of SimpleRanker

IDF Ranker

Example

gfmrag/workflow/config/doc_ranker/idf_ranker.yaml
_target_: gfmrag.doc_rankers.IDFWeightedRanker
Parameter Options Note
_target_ gfmrag.doc_rankers.IDFWeightedRanker The class name of IDFWeightedRanker

Top-k Ranker

Example

gfmrag/workflow/config/doc_ranker/topk_ranker.yaml
_target_: gfmrag.doc_rankers.TopKRanker
top_k: 10
Parameter Options Note
_target_ gfmrag.doc_rankers.TopKRanker The class name of TopKRanker
top_k Integer The top-k entities used for document retrieval

IDF Top-k Ranker

Example

gfmrag/workflow/config/doc_ranker/idf_topk_ranker.yaml
_target_: gfmrag.doc_rankers.IDFWeightedTopKRanker
top_k: 20
Parameter Options Note
_target_ gfmrag.doc_rankers.IDFWeightedTopKRanker The class name of IDFWeightedTopKRanker
top_k Integer The top-k entities used for document retrieval