config_wn.json

{
    "batch_size": 100,
    "data_dir": "data/WN18RR",
    "task": "WN18RR",
    "bge_model": "/home/kayla/models/bge-reranker-v2-m3",
    "relation_path": "data/WN18RR/relations.json",
    "entity_desc_path": "data/WN18RR/entities.json",
    "relation_template_path": "data/WN18RR/relations2_finalTemplate.json",
    "train_path": "data/WN18RR/train.txt.json",
    "train_txt_path": "data/WN18RR/train.txt",
    "test_txt_path": "data/WN18RR/test.txt",
    "fact_path": "data/WN18RR/valid.txt.json",
    "test_dir": "data/WN18RR/small_test",
    "new_test_path": "data/WN18RR/new_test.txt.json",
    "entity_name2id_path": "data/WN18RR/entity_name2id.json",
    "entity_id2name_path": "data/WN18RR/entity_id2name.json",
    "save_neighbors_path": "data/WN18RR/neighbors.txt",
    "retriver_dir": "data/WN18RR/retrieval",
    "LLM_result_dir": "data/WN18RR/llm_answers/select",
    "new_rank_result_dir": "data/WN18RR/llm_answers/new_rerank",
    "use_link_graph": true,
    "workers": 0,
    "nei_limitation": 8,
    "wiki_limitation": 2,
    "llm_model": "gpt-4o-mini",
    "eval_llm_model": "gpt-4o",
    "use_filter_candidates": true,
    "use_relation_template": true,
    "use_kg_neis": true,
    "use_wiki_docs": true,
    "image_dir": "data/WN18RR/images/test",
    "entity_frequency_path": "data/WN18RR/entity_frequency.json",
    "long_tail_dir": "data/WN18RR/long_tail_data",
    "long_tail_degree_dir": "data/WN18RR/long_tail_data_degree",
    "eval_type": "llm",
    "transE_dir": "data/WN18RR/TransE",
    "simkgc_dir": "data/WN18RR/SimKGC",
    "long_tail_image_dir": "small_experiments/long_tail/WN18RR/images",
    "long_tail_image_degree_dir": "small_experiments/long_tail/WN18RR/images_degree",
    "select_query_dir": "data/WN18RR/select_query",
    "degree": false,
    "evaluation_dir": "data/WN18RR/evaluation"
}