Hugging Face's logo

rmanluo
/
GFM-RAG-8M

Model card Files
xet
 Community
GFM-RAG-8M
File size: 1,698 Bytes
15d1a73
 
 
 
 
 
 
 
 
 
 
4da9e46
 
 
 
 
 
 
 
15d1a73
4da9e46
15d1a73
 
 
 
 
 
 
 
 
 
 
 
 
4da9e46
 
 
 
 
 
 
 
 
 
 
 
15d1a73
4da9e46
 
 
 
 
15d1a73
4da9e46
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
{
    "text_emb_model_config": {
        "_target_": "gfmrag.text_emb_models.BaseTextEmbModel",
        "text_emb_model_name": "sentence-transformers/all-mpnet-base-v2",
        "normalize": false,
        "batch_size": 32,
        "query_instruct": null,
        "passage_instruct": null,
        "model_kwargs": null
    },
    "model_config": {
        "_target_": "gfmrag.models.gfm_rag_v1.GNNRetriever",
        "feat_dim": 768,
        "init_nodes_weight": true,
        "init_nodes_type": "document",
        "ranker": {
            "_target_": "gfmrag.models.gfm_rag_v1.rankers.IDFWeightedTopKRanker",
            "top_k": 20
        },
        "entity_model": {
            "_target_": "gfmrag.models.ultra.models.QueryNBFNet",
            "input_dim": 512,
            "hidden_dims": [
                512,
                512,
                512,
                512,
                512,
                512
            ],
            "message_func": "distmult",
            "aggregate_func": "sum",
            "short_cut": true,
            "layer_norm": true
        }
    },
    "dataset_config": {
        "class_name": "GraphIndexDatasetV1",
        "text_emb_model_cfgs": {
            "_target_": "gfmrag.text_emb_models.BaseTextEmbModel",
            "text_emb_model_name": "sentence-transformers/all-mpnet-base-v2",
            "normalize": false,
            "batch_size": 32,
            "query_instruct": null,
            "passage_instruct": null,
            "model_kwargs": null
        },
        "target_type": "entity",
        "use_node_feat": false,
        "use_relation_feat": true,
        "use_edge_feat": false,
        "inverse_relation_feat": "text"
    }
}