Document Ranker Configuration

Simple Ranker

An example Simple Ranker configuration file is shown below:

simple_ranker

gfmrag/workflow/config/doc_ranker/simple_ranker.yaml

_target_: gfmrag.models.gfm_rag_v1.rankers.SimpleRanker

Parameter	Options	Note
_target_	gfmrag.models.gfm_rag_v1.rankers.SimpleRanker	The class name of SimpleRanker

IDF Ranker

An example IDF Ranker configuration file is shown below:

idf_ranker

gfmrag/workflow/config/doc_ranker/idf_ranker.yaml

_target_: gfmrag.models.gfm_rag_v1.rankers.IDFWeightedRanker

Parameter	Options	Note
_target_	gfmrag.models.gfm_rag_v1.rankers.IDFWeightedRanker	The class name of IDFWeightedRanker

Top-k Ranker

An example Top-k Ranker configuration file is shown below:

topk_ranker

gfmrag/workflow/config/doc_ranker/topk_ranker.yaml

_target_: gfmrag.models.gfm_rag_v1.rankers.TopKRanker
top_k: 10

Parameter	Options	Note
_target_	gfmrag.models.gfm_rag_v1.rankers.TopKRanker	The class name of TopKRanker
top_k	Integer	The top-k entities used for document retrieval

IDF Top-k Ranker

An example IDF Top-k Ranker configuration file is shown below:

idf_topk_ranker

gfmrag/workflow/config/doc_ranker/idf_topk_ranker.yaml

_target_: gfmrag.models.gfm_rag_v1.rankers.IDFWeightedTopKRanker
top_k: 20

Parameter	Options	Note
_target_	gfmrag.models.gfm_rag_v1.rankers.IDFWeightedTopKRanker	The class name of IDFWeightedTopKRanker
top_k	Integer	The top-k entities used for document retrieval