├── .gitignore ├── README.md ├── run.sh ├── scripts ├── eval_graftnet.sh ├── eval_nsm.sh ├── eval_nsm_ppr.sh ├── eval_pullnet.sh ├── eval_retriever.sh ├── run_train_retriever.sh └── run_unsup_train_retriever.sh └── src ├── CWQ ├── SentEval │ ├── .gitignore │ ├── LICENSE │ ├── README.md │ ├── examples │ │ ├── bow.py │ │ ├── gensen.py │ │ ├── googleuse.py │ │ ├── infersent.py │ │ └── skipthought.py │ ├── senteval │ │ ├── __init__.py │ │ ├── binary.py │ │ ├── engine.py │ │ ├── mrpc.py │ │ ├── probing.py │ │ ├── rank.py │ │ ├── sick.py │ │ ├── snli.py │ │ ├── sst.py │ │ ├── sts.py │ │ ├── tools │ │ │ ├── __init__.py │ │ │ ├── classifier.py │ │ │ ├── ranking.py │ │ │ ├── relatedness.py │ │ │ └── validation.py │ │ ├── trec.py │ │ └── utils.py │ └── setup.py ├── config.py ├── end2end_graftnet │ ├── LICENSE │ ├── README.md │ ├── config │ │ └── webqsp_e2e_generate.yml │ ├── data_loader.py │ ├── environment.yml │ ├── graftnet.py │ ├── main.py │ ├── retrieve_paths │ │ ├── build_relation_set_for_end2end.py │ │ └── retrieve_subgraph_for_end2end.py │ ├── run_retrieve_subgraph.py │ ├── score_paths.sh │ ├── script.py │ └── util.py ├── end2end_nsm │ ├── NSM │ │ ├── Agent │ │ │ ├── BaseAgent.py │ │ │ ├── NSMAgent.py │ │ │ ├── TeacherAgent.py │ │ │ └── TeacherAgent2.py │ │ ├── Modules │ │ │ ├── Instruction │ │ │ │ ├── base_instruction.py │ │ │ │ └── seq_instruction.py │ │ │ ├── Reasoning │ │ │ │ ├── base_reasoning.py │ │ │ │ ├── gnn_backward_reasoning.py │ │ │ │ └── gnn_reasoning.py │ │ │ ├── layer_nsm.py │ │ │ └── layers_att.py │ │ ├── train │ │ │ ├── evaluate_nsm.py │ │ │ ├── init.py │ │ │ ├── trainer_hybrid.py │ │ │ ├── trainer_nsm.py │ │ │ ├── trainer_parallel.py │ │ │ └── trainer_student.py │ │ └── util │ │ │ ├── config.py │ │ │ └── utils.py │ ├── finetune_preprocess │ │ ├── build_search_tree_for_cwq_dataset.py │ │ ├── build_search_tree_for_dataset.py │ │ ├── build_train_example.py │ │ └── cal_path_for_cwq.py │ ├── main_nsm.py │ ├── retrieve_paths │ │ ├── build_relation_set_for_end2end.py │ │ └── retrieve_subgraph_for_end2end.py │ ├── run_retrieve_subgraph.py │ └── score_paths.sh ├── generate_kb_cache │ ├── build_ent_type_ary.py │ ├── convert_all_triple_to_int.py │ └── map_test_set.py ├── graftnet_reader │ ├── LICENSE │ ├── README.md │ ├── config │ │ ├── webqsp.yml │ │ ├── webqsp.yml.bak │ │ ├── webqsp_e2e_generate.yml │ │ ├── webqsp_eval.yml │ │ └── wikimovie.yml │ ├── data_loader.py │ ├── environment.yml │ ├── graftnet.py │ ├── main.py │ ├── pred_example.json │ ├── preprocessing │ │ ├── README.md │ │ ├── run_pipeline │ │ ├── step0_preprocess_webqsp.py │ │ ├── step1_process_entity_links.py │ │ ├── step2_relation_embeddings.py │ │ ├── step3_question_embeddings.py │ │ └── step4_extract_subgraphs.py │ ├── script.py │ └── util.py ├── knowledge_graph │ ├── knowledge_graph.py │ ├── knowledge_graph_base.py │ ├── knowledge_graph_cache.py │ └── knowledge_graph_freebase.py ├── nsm_reader │ ├── .gitignore │ ├── NSM │ │ ├── Agent │ │ │ ├── BaseAgent.py │ │ │ ├── NSMAgent.py │ │ │ ├── TeacherAgent.py │ │ │ └── TeacherAgent2.py │ │ ├── Modules │ │ │ ├── Instruction │ │ │ │ ├── base_instruction.py │ │ │ │ └── seq_instruction.py │ │ │ ├── Reasoning │ │ │ │ ├── base_reasoning.py │ │ │ │ ├── gnn_backward_reasoning.py │ │ │ │ └── gnn_reasoning.py │ │ │ ├── layer_nsm.py │ │ │ └── layers_att.py │ │ ├── train │ │ │ ├── evaluate_nsm.py │ │ │ ├── init.py │ │ │ ├── trainer_hybrid.py │ │ │ ├── trainer_nsm.py │ │ │ ├── trainer_parallel.py │ │ │ └── trainer_student.py │ │ └── util │ │ │ ├── config.py │ │ │ └── utils.py │ ├── eval.sh │ └── main_nsm.py ├── preprocessing │ ├── build_training_dataset.py │ ├── filter_legal_path.py │ ├── negative_sampling.py │ ├── score_path.py │ └── search_to_get_path.py ├── retrieve_subgraph │ ├── build_relation_set.py │ ├── eval.py │ ├── generate_relation_cache.py │ ├── retrieve_subgraph.py │ └── utils.py ├── retriever │ ├── README.md │ ├── convert_to_plm.py │ ├── generate_tokenized_data.py │ ├── model.py │ ├── retriever_2 │ │ ├── convert_to_plm.py │ │ ├── model.py │ │ ├── tokenize_csv_file.py │ │ ├── tokenize_jsonl_file.py │ │ ├── train.py │ │ ├── train_full.sh │ │ ├── trainer.py │ │ └── utils.py │ ├── train.py │ ├── train.sh │ ├── trainer.py │ └── utils.py ├── run_convert_retriever_output_to_graftnet.py ├── run_convert_retriever_output_to_graftnet_end2end.py ├── run_end2end_graftnet.sh ├── run_end2end_nsm.sh ├── run_preprocess.py ├── run_retrieve_subgraph.py ├── run_train_graftnet.sh ├── run_train_nsm.py ├── run_train_nsm.sh └── utils.py ├── SentEval ├── .gitignore ├── LICENSE ├── README.md ├── examples │ ├── bow.py │ ├── gensen.py │ ├── googleuse.py │ ├── infersent.py │ └── skipthought.py ├── senteval │ ├── __init__.py │ ├── binary.py │ ├── engine.py │ ├── mrpc.py │ ├── probing.py │ ├── rank.py │ ├── sick.py │ ├── snli.py │ ├── sst.py │ ├── sts.py │ ├── tools │ │ ├── __init__.py │ │ ├── classifier.py │ │ ├── ranking.py │ │ ├── relatedness.py │ │ └── validation.py │ ├── trec.py │ └── utils.py └── setup.py ├── config.py ├── graftnet_reader ├── LICENSE ├── README.md ├── config │ ├── webqsp.yml │ ├── webqsp.yml.bak │ ├── webqsp_test.yml │ └── wikimovie.yml ├── data_loader.py ├── environment.yml ├── graftnet.py ├── load_data_from_nsm.py ├── main.py ├── pred_example.json ├── preprocessing │ ├── README.md │ ├── run_pipeline │ ├── step0_preprocess_webqsp.py │ ├── step1_process_entity_links.py │ ├── step2_relation_embeddings.py │ ├── step3_question_embeddings.py │ └── step4_extract_subgraphs.py ├── script.py └── util.py ├── knowledge_graph ├── knowledge_graph.py ├── knowledge_graph_cache.py └── knowledge_graph_freebase.py ├── nsm_reader ├── .gitignore ├── NSM │ ├── Agent │ │ ├── BaseAgent.py │ │ ├── NSMAgent.py │ │ ├── TeacherAgent.py │ │ └── TeacherAgent2.py │ ├── Model │ │ ├── backward_model.py │ │ ├── base_model.py │ │ ├── forward_model.py │ │ ├── hybrid_model.py │ │ └── nsm_model.py │ ├── Modules │ │ ├── Instruction │ │ │ ├── base_instruction.py │ │ │ └── seq_instruction.py │ │ ├── Reasoning │ │ │ ├── base_reasoning.py │ │ │ ├── gnn_backward_reasoning.py │ │ │ └── gnn_reasoning.py │ │ ├── layer_nsm.py │ │ └── layers_att.py │ ├── data │ │ ├── basic_dataset.py │ │ ├── dataset_single.py │ │ ├── dataset_super.py │ │ ├── load_data.py │ │ ├── load_data_super.py │ │ ├── read_tree.py │ │ └── util.py │ ├── train │ │ ├── evaluate_nsm.py │ │ ├── init.py │ │ ├── trainer_hybrid.py │ │ ├── trainer_nsm.py │ │ ├── trainer_parallel.py │ │ └── trainer_student.py │ └── util │ │ ├── config.py │ │ └── utils.py ├── main_nsm.py ├── preprocessing │ ├── .DS_Store │ ├── Freebase │ │ ├── .DS_Store │ │ ├── README.md │ │ ├── build_vocab_from_dep.py │ │ ├── get_2hop_subgraph.py │ │ ├── get_seed_set.py │ │ ├── load_emb_glove.py │ │ ├── manual_filter_rel.py │ │ ├── map_kb_id.py │ │ ├── preprocess_step0.py │ │ ├── preprocess_step1.py │ │ ├── update_vocab_with_rel.py │ │ └── util │ │ │ ├── deal_cvt.py │ │ │ └── ppr_util.py │ ├── MetaQA │ │ ├── README.md │ │ ├── build_vocab_from_dep.py │ │ ├── load_emb_glove.py │ │ ├── map_kb_id.py │ │ ├── ppr_util.py │ │ ├── prepare_data.py │ │ └── simplify_dataset.py │ └── parse │ │ ├── con_parse.py │ │ ├── const_parse.sh │ │ ├── dep_parse.py │ │ ├── dependency_parse.sh │ │ ├── readme.txt │ │ └── server.sh └── presentation │ ├── wsdm-poster.pdf │ └── wsdm_slides_ver2.pptx ├── preprocessing ├── load_dataset.py ├── negative_sampling.py ├── negative_sampling_for_unsup.py ├── score_path.py └── search_to_get_path.py ├── retrieve_subgraph ├── build_relation_set.py ├── retrieve_subgraph.py ├── retrieve_subgraph_for_finetune.py ├── retrieve_subgraph_for_graftnet.py ├── retrieve_subgraph_for_test.py └── utils.py ├── retriever ├── LICENSE ├── evaluation.py ├── huggingface_to_simcse.py ├── requirements.txt ├── simcse_to_huggingface.py ├── src │ ├── models.py │ └── trainers.py └── train.py ├── retriever_finetune ├── NSM │ ├── Agent │ │ ├── BaseAgent.py │ │ ├── NSMAgent.py │ │ ├── TeacherAgent.py │ │ └── TeacherAgent2.py │ ├── Model │ │ ├── backward_model.py │ │ ├── base_model.py │ │ ├── forward_model.py │ │ ├── hybrid_model.py │ │ └── nsm_model.py │ ├── Modules │ │ ├── Instruction │ │ │ ├── base_instruction.py │ │ │ └── seq_instruction.py │ │ ├── Reasoning │ │ │ ├── base_reasoning.py │ │ │ ├── gnn_backward_reasoning.py │ │ │ └── gnn_reasoning.py │ │ ├── layer_nsm.py │ │ └── layers_att.py │ ├── data │ │ ├── basic_dataset.py │ │ ├── dataset_single.py │ │ ├── dataset_super.py │ │ ├── load_data.py │ │ ├── load_data_super.py │ │ ├── read_tree.py │ │ └── util.py │ ├── train │ │ ├── evaluate_nsm.py │ │ ├── init.py │ │ ├── trainer_hybrid.py │ │ ├── trainer_nsm.py │ │ ├── trainer_parallel.py │ │ └── trainer_student.py │ └── util │ │ ├── config.py │ │ └── utils.py ├── build_search_tree_for_dataset.py ├── build_train_set_from_search_tree.py ├── cal_path_score_from_reader.py ├── m_dict.json ├── main_nsm.py ├── search_tree_list.pkl └── utils.py ├── run_preprocess.py ├── run_retrieve_subgraph.py ├── run_retriever_finetune.py ├── run_retriever_finetune.sh ├── run_train_graftnet.py ├── run_train_graftnet.sh ├── run_train_nsm.py ├── run_train_nsm.sh ├── run_train_retriever.py ├── run_train_retriever.sh └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/README.md -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/run.sh -------------------------------------------------------------------------------- /scripts/eval_graftnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/scripts/eval_graftnet.sh -------------------------------------------------------------------------------- /scripts/eval_nsm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/scripts/eval_nsm.sh -------------------------------------------------------------------------------- /scripts/eval_nsm_ppr.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/scripts/eval_nsm_ppr.sh -------------------------------------------------------------------------------- /scripts/eval_pullnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/scripts/eval_pullnet.sh -------------------------------------------------------------------------------- /scripts/eval_retriever.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/scripts/eval_retriever.sh -------------------------------------------------------------------------------- /scripts/run_train_retriever.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/scripts/run_train_retriever.sh -------------------------------------------------------------------------------- /scripts/run_unsup_train_retriever.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/scripts/run_unsup_train_retriever.sh -------------------------------------------------------------------------------- /src/CWQ/SentEval/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/.gitignore -------------------------------------------------------------------------------- /src/CWQ/SentEval/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/LICENSE -------------------------------------------------------------------------------- /src/CWQ/SentEval/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/README.md -------------------------------------------------------------------------------- /src/CWQ/SentEval/examples/bow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/examples/bow.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/examples/gensen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/examples/gensen.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/examples/googleuse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/examples/googleuse.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/examples/infersent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/examples/infersent.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/examples/skipthought.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/examples/skipthought.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/__init__.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/binary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/binary.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/engine.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/mrpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/mrpc.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/probing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/probing.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/rank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/rank.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/sick.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/sick.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/snli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/snli.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/sst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/sst.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/sts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/sts.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/tools/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/tools/classifier.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/tools/ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/tools/ranking.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/tools/relatedness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/tools/relatedness.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/tools/validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/tools/validation.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/trec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/trec.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/senteval/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/senteval/utils.py -------------------------------------------------------------------------------- /src/CWQ/SentEval/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/SentEval/setup.py -------------------------------------------------------------------------------- /src/CWQ/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/config.py -------------------------------------------------------------------------------- /src/CWQ/end2end_graftnet/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_graftnet/LICENSE -------------------------------------------------------------------------------- /src/CWQ/end2end_graftnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_graftnet/README.md -------------------------------------------------------------------------------- /src/CWQ/end2end_graftnet/config/webqsp_e2e_generate.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_graftnet/config/webqsp_e2e_generate.yml -------------------------------------------------------------------------------- /src/CWQ/end2end_graftnet/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_graftnet/data_loader.py -------------------------------------------------------------------------------- /src/CWQ/end2end_graftnet/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_graftnet/environment.yml -------------------------------------------------------------------------------- /src/CWQ/end2end_graftnet/graftnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_graftnet/graftnet.py -------------------------------------------------------------------------------- /src/CWQ/end2end_graftnet/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_graftnet/main.py -------------------------------------------------------------------------------- /src/CWQ/end2end_graftnet/retrieve_paths/build_relation_set_for_end2end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_graftnet/retrieve_paths/build_relation_set_for_end2end.py -------------------------------------------------------------------------------- /src/CWQ/end2end_graftnet/retrieve_paths/retrieve_subgraph_for_end2end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_graftnet/retrieve_paths/retrieve_subgraph_for_end2end.py -------------------------------------------------------------------------------- /src/CWQ/end2end_graftnet/run_retrieve_subgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_graftnet/run_retrieve_subgraph.py -------------------------------------------------------------------------------- /src/CWQ/end2end_graftnet/score_paths.sh: -------------------------------------------------------------------------------- 1 | python main.py --test config/webqsp_e2e_generate.yml 2 | -------------------------------------------------------------------------------- /src/CWQ/end2end_graftnet/script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_graftnet/script.py -------------------------------------------------------------------------------- /src/CWQ/end2end_graftnet/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_graftnet/util.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/Agent/BaseAgent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/Agent/BaseAgent.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/Agent/NSMAgent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/Agent/NSMAgent.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/Agent/TeacherAgent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/Agent/TeacherAgent.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/Agent/TeacherAgent2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/Agent/TeacherAgent2.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/Modules/Instruction/base_instruction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/Modules/Instruction/base_instruction.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/Modules/Instruction/seq_instruction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/Modules/Instruction/seq_instruction.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/Modules/Reasoning/base_reasoning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/Modules/Reasoning/base_reasoning.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/Modules/Reasoning/gnn_backward_reasoning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/Modules/Reasoning/gnn_backward_reasoning.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/Modules/Reasoning/gnn_reasoning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/Modules/Reasoning/gnn_reasoning.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/Modules/layer_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/Modules/layer_nsm.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/Modules/layers_att.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/Modules/layers_att.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/train/evaluate_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/train/evaluate_nsm.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/train/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/train/init.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/train/trainer_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/train/trainer_hybrid.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/train/trainer_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/train/trainer_nsm.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/train/trainer_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/train/trainer_parallel.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/train/trainer_student.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/train/trainer_student.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/util/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/util/config.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/NSM/util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/NSM/util/utils.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/finetune_preprocess/build_search_tree_for_cwq_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/finetune_preprocess/build_search_tree_for_cwq_dataset.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/finetune_preprocess/build_search_tree_for_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/finetune_preprocess/build_search_tree_for_dataset.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/finetune_preprocess/build_train_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/finetune_preprocess/build_train_example.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/finetune_preprocess/cal_path_for_cwq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/finetune_preprocess/cal_path_for_cwq.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/main_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/main_nsm.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/retrieve_paths/build_relation_set_for_end2end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/retrieve_paths/build_relation_set_for_end2end.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/retrieve_paths/retrieve_subgraph_for_end2end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/retrieve_paths/retrieve_subgraph_for_end2end.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/run_retrieve_subgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/run_retrieve_subgraph.py -------------------------------------------------------------------------------- /src/CWQ/end2end_nsm/score_paths.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/end2end_nsm/score_paths.sh -------------------------------------------------------------------------------- /src/CWQ/generate_kb_cache/build_ent_type_ary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/generate_kb_cache/build_ent_type_ary.py -------------------------------------------------------------------------------- /src/CWQ/generate_kb_cache/convert_all_triple_to_int.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/generate_kb_cache/convert_all_triple_to_int.py -------------------------------------------------------------------------------- /src/CWQ/generate_kb_cache/map_test_set.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/generate_kb_cache/map_test_set.py -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/LICENSE -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/README.md -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/config/webqsp.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/config/webqsp.yml -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/config/webqsp.yml.bak: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/config/webqsp.yml.bak -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/config/webqsp_e2e_generate.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/config/webqsp_e2e_generate.yml -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/config/webqsp_eval.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/config/webqsp_eval.yml -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/config/wikimovie.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/config/wikimovie.yml -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/data_loader.py -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/environment.yml -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/graftnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/graftnet.py -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/main.py -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/pred_example.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/pred_example.json -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/preprocessing/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/preprocessing/README.md -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/preprocessing/run_pipeline: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/preprocessing/run_pipeline -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/preprocessing/step0_preprocess_webqsp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/preprocessing/step0_preprocess_webqsp.py -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/preprocessing/step1_process_entity_links.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/preprocessing/step1_process_entity_links.py -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/preprocessing/step2_relation_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/preprocessing/step2_relation_embeddings.py -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/preprocessing/step3_question_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/preprocessing/step3_question_embeddings.py -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/preprocessing/step4_extract_subgraphs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/preprocessing/step4_extract_subgraphs.py -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/script.py -------------------------------------------------------------------------------- /src/CWQ/graftnet_reader/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/graftnet_reader/util.py -------------------------------------------------------------------------------- /src/CWQ/knowledge_graph/knowledge_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/knowledge_graph/knowledge_graph.py -------------------------------------------------------------------------------- /src/CWQ/knowledge_graph/knowledge_graph_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/knowledge_graph/knowledge_graph_base.py -------------------------------------------------------------------------------- /src/CWQ/knowledge_graph/knowledge_graph_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/knowledge_graph/knowledge_graph_cache.py -------------------------------------------------------------------------------- /src/CWQ/knowledge_graph/knowledge_graph_freebase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/knowledge_graph/knowledge_graph_freebase.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/.gitignore -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/Agent/BaseAgent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/Agent/BaseAgent.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/Agent/NSMAgent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/Agent/NSMAgent.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/Agent/TeacherAgent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/Agent/TeacherAgent.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/Agent/TeacherAgent2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/Agent/TeacherAgent2.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/Modules/Instruction/base_instruction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/Modules/Instruction/base_instruction.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/Modules/Instruction/seq_instruction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/Modules/Instruction/seq_instruction.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/Modules/Reasoning/base_reasoning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/Modules/Reasoning/base_reasoning.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/Modules/Reasoning/gnn_backward_reasoning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/Modules/Reasoning/gnn_backward_reasoning.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/Modules/Reasoning/gnn_reasoning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/Modules/Reasoning/gnn_reasoning.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/Modules/layer_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/Modules/layer_nsm.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/Modules/layers_att.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/Modules/layers_att.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/train/evaluate_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/train/evaluate_nsm.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/train/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/train/init.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/train/trainer_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/train/trainer_hybrid.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/train/trainer_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/train/trainer_nsm.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/train/trainer_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/train/trainer_parallel.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/train/trainer_student.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/train/trainer_student.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/util/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/util/config.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/NSM/util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/NSM/util/utils.py -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/eval.sh -------------------------------------------------------------------------------- /src/CWQ/nsm_reader/main_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/nsm_reader/main_nsm.py -------------------------------------------------------------------------------- /src/CWQ/preprocessing/build_training_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/preprocessing/build_training_dataset.py -------------------------------------------------------------------------------- /src/CWQ/preprocessing/filter_legal_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/preprocessing/filter_legal_path.py -------------------------------------------------------------------------------- /src/CWQ/preprocessing/negative_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/preprocessing/negative_sampling.py -------------------------------------------------------------------------------- /src/CWQ/preprocessing/score_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/preprocessing/score_path.py -------------------------------------------------------------------------------- /src/CWQ/preprocessing/search_to_get_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/preprocessing/search_to_get_path.py -------------------------------------------------------------------------------- /src/CWQ/retrieve_subgraph/build_relation_set.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retrieve_subgraph/build_relation_set.py -------------------------------------------------------------------------------- /src/CWQ/retrieve_subgraph/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retrieve_subgraph/eval.py -------------------------------------------------------------------------------- /src/CWQ/retrieve_subgraph/generate_relation_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retrieve_subgraph/generate_relation_cache.py -------------------------------------------------------------------------------- /src/CWQ/retrieve_subgraph/retrieve_subgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retrieve_subgraph/retrieve_subgraph.py -------------------------------------------------------------------------------- /src/CWQ/retrieve_subgraph/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retrieve_subgraph/utils.py -------------------------------------------------------------------------------- /src/CWQ/retriever/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/README.md -------------------------------------------------------------------------------- /src/CWQ/retriever/convert_to_plm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/convert_to_plm.py -------------------------------------------------------------------------------- /src/CWQ/retriever/generate_tokenized_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/generate_tokenized_data.py -------------------------------------------------------------------------------- /src/CWQ/retriever/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/model.py -------------------------------------------------------------------------------- /src/CWQ/retriever/retriever_2/convert_to_plm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/retriever_2/convert_to_plm.py -------------------------------------------------------------------------------- /src/CWQ/retriever/retriever_2/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/retriever_2/model.py -------------------------------------------------------------------------------- /src/CWQ/retriever/retriever_2/tokenize_csv_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/retriever_2/tokenize_csv_file.py -------------------------------------------------------------------------------- /src/CWQ/retriever/retriever_2/tokenize_jsonl_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/retriever_2/tokenize_jsonl_file.py -------------------------------------------------------------------------------- /src/CWQ/retriever/retriever_2/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/retriever_2/train.py -------------------------------------------------------------------------------- /src/CWQ/retriever/retriever_2/train_full.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/retriever_2/train_full.sh -------------------------------------------------------------------------------- /src/CWQ/retriever/retriever_2/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/retriever_2/trainer.py -------------------------------------------------------------------------------- /src/CWQ/retriever/retriever_2/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/retriever_2/utils.py -------------------------------------------------------------------------------- /src/CWQ/retriever/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/train.py -------------------------------------------------------------------------------- /src/CWQ/retriever/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/train.sh -------------------------------------------------------------------------------- /src/CWQ/retriever/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/trainer.py -------------------------------------------------------------------------------- /src/CWQ/retriever/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/retriever/utils.py -------------------------------------------------------------------------------- /src/CWQ/run_convert_retriever_output_to_graftnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/run_convert_retriever_output_to_graftnet.py -------------------------------------------------------------------------------- /src/CWQ/run_convert_retriever_output_to_graftnet_end2end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/run_convert_retriever_output_to_graftnet_end2end.py -------------------------------------------------------------------------------- /src/CWQ/run_end2end_graftnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/run_end2end_graftnet.sh -------------------------------------------------------------------------------- /src/CWQ/run_end2end_nsm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/run_end2end_nsm.sh -------------------------------------------------------------------------------- /src/CWQ/run_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/run_preprocess.py -------------------------------------------------------------------------------- /src/CWQ/run_retrieve_subgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/run_retrieve_subgraph.py -------------------------------------------------------------------------------- /src/CWQ/run_train_graftnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/run_train_graftnet.sh -------------------------------------------------------------------------------- /src/CWQ/run_train_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/run_train_nsm.py -------------------------------------------------------------------------------- /src/CWQ/run_train_nsm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/run_train_nsm.sh -------------------------------------------------------------------------------- /src/CWQ/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/CWQ/utils.py -------------------------------------------------------------------------------- /src/SentEval/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/.gitignore -------------------------------------------------------------------------------- /src/SentEval/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/LICENSE -------------------------------------------------------------------------------- /src/SentEval/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/README.md -------------------------------------------------------------------------------- /src/SentEval/examples/bow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/examples/bow.py -------------------------------------------------------------------------------- /src/SentEval/examples/gensen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/examples/gensen.py -------------------------------------------------------------------------------- /src/SentEval/examples/googleuse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/examples/googleuse.py -------------------------------------------------------------------------------- /src/SentEval/examples/infersent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/examples/infersent.py -------------------------------------------------------------------------------- /src/SentEval/examples/skipthought.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/examples/skipthought.py -------------------------------------------------------------------------------- /src/SentEval/senteval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/__init__.py -------------------------------------------------------------------------------- /src/SentEval/senteval/binary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/binary.py -------------------------------------------------------------------------------- /src/SentEval/senteval/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/engine.py -------------------------------------------------------------------------------- /src/SentEval/senteval/mrpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/mrpc.py -------------------------------------------------------------------------------- /src/SentEval/senteval/probing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/probing.py -------------------------------------------------------------------------------- /src/SentEval/senteval/rank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/rank.py -------------------------------------------------------------------------------- /src/SentEval/senteval/sick.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/sick.py -------------------------------------------------------------------------------- /src/SentEval/senteval/snli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/snli.py -------------------------------------------------------------------------------- /src/SentEval/senteval/sst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/sst.py -------------------------------------------------------------------------------- /src/SentEval/senteval/sts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/sts.py -------------------------------------------------------------------------------- /src/SentEval/senteval/tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/SentEval/senteval/tools/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/tools/classifier.py -------------------------------------------------------------------------------- /src/SentEval/senteval/tools/ranking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/tools/ranking.py -------------------------------------------------------------------------------- /src/SentEval/senteval/tools/relatedness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/tools/relatedness.py -------------------------------------------------------------------------------- /src/SentEval/senteval/tools/validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/tools/validation.py -------------------------------------------------------------------------------- /src/SentEval/senteval/trec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/trec.py -------------------------------------------------------------------------------- /src/SentEval/senteval/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/senteval/utils.py -------------------------------------------------------------------------------- /src/SentEval/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/SentEval/setup.py -------------------------------------------------------------------------------- /src/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/config.py -------------------------------------------------------------------------------- /src/graftnet_reader/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/LICENSE -------------------------------------------------------------------------------- /src/graftnet_reader/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/README.md -------------------------------------------------------------------------------- /src/graftnet_reader/config/webqsp.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/config/webqsp.yml -------------------------------------------------------------------------------- /src/graftnet_reader/config/webqsp.yml.bak: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/config/webqsp.yml.bak -------------------------------------------------------------------------------- /src/graftnet_reader/config/webqsp_test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/config/webqsp_test.yml -------------------------------------------------------------------------------- /src/graftnet_reader/config/wikimovie.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/config/wikimovie.yml -------------------------------------------------------------------------------- /src/graftnet_reader/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/data_loader.py -------------------------------------------------------------------------------- /src/graftnet_reader/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/environment.yml -------------------------------------------------------------------------------- /src/graftnet_reader/graftnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/graftnet.py -------------------------------------------------------------------------------- /src/graftnet_reader/load_data_from_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/load_data_from_nsm.py -------------------------------------------------------------------------------- /src/graftnet_reader/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/main.py -------------------------------------------------------------------------------- /src/graftnet_reader/pred_example.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/pred_example.json -------------------------------------------------------------------------------- /src/graftnet_reader/preprocessing/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/preprocessing/README.md -------------------------------------------------------------------------------- /src/graftnet_reader/preprocessing/run_pipeline: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/preprocessing/run_pipeline -------------------------------------------------------------------------------- /src/graftnet_reader/preprocessing/step0_preprocess_webqsp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/preprocessing/step0_preprocess_webqsp.py -------------------------------------------------------------------------------- /src/graftnet_reader/preprocessing/step1_process_entity_links.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/preprocessing/step1_process_entity_links.py -------------------------------------------------------------------------------- /src/graftnet_reader/preprocessing/step2_relation_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/preprocessing/step2_relation_embeddings.py -------------------------------------------------------------------------------- /src/graftnet_reader/preprocessing/step3_question_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/preprocessing/step3_question_embeddings.py -------------------------------------------------------------------------------- /src/graftnet_reader/preprocessing/step4_extract_subgraphs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/preprocessing/step4_extract_subgraphs.py -------------------------------------------------------------------------------- /src/graftnet_reader/script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/script.py -------------------------------------------------------------------------------- /src/graftnet_reader/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/graftnet_reader/util.py -------------------------------------------------------------------------------- /src/knowledge_graph/knowledge_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/knowledge_graph/knowledge_graph.py -------------------------------------------------------------------------------- /src/knowledge_graph/knowledge_graph_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/knowledge_graph/knowledge_graph_cache.py -------------------------------------------------------------------------------- /src/knowledge_graph/knowledge_graph_freebase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/knowledge_graph/knowledge_graph_freebase.py -------------------------------------------------------------------------------- /src/nsm_reader/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/.gitignore -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Agent/BaseAgent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Agent/BaseAgent.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Agent/NSMAgent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Agent/NSMAgent.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Agent/TeacherAgent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Agent/TeacherAgent.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Agent/TeacherAgent2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Agent/TeacherAgent2.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Model/backward_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Model/backward_model.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Model/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Model/base_model.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Model/forward_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Model/forward_model.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Model/hybrid_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Model/hybrid_model.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Model/nsm_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Model/nsm_model.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Modules/Instruction/base_instruction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Modules/Instruction/base_instruction.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Modules/Instruction/seq_instruction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Modules/Instruction/seq_instruction.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Modules/Reasoning/base_reasoning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Modules/Reasoning/base_reasoning.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Modules/Reasoning/gnn_backward_reasoning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Modules/Reasoning/gnn_backward_reasoning.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Modules/Reasoning/gnn_reasoning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Modules/Reasoning/gnn_reasoning.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Modules/layer_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Modules/layer_nsm.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/Modules/layers_att.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/Modules/layers_att.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/data/basic_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/data/basic_dataset.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/data/dataset_single.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/data/dataset_single.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/data/dataset_super.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/data/dataset_super.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/data/load_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/data/load_data.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/data/load_data_super.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/data/load_data_super.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/data/read_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/data/read_tree.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/data/util.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/train/evaluate_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/train/evaluate_nsm.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/train/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/train/init.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/train/trainer_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/train/trainer_hybrid.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/train/trainer_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/train/trainer_nsm.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/train/trainer_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/train/trainer_parallel.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/train/trainer_student.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/train/trainer_student.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/util/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/util/config.py -------------------------------------------------------------------------------- /src/nsm_reader/NSM/util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/NSM/util/utils.py -------------------------------------------------------------------------------- /src/nsm_reader/main_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/main_nsm.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/.DS_Store -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/Freebase/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/Freebase/.DS_Store -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/Freebase/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/Freebase/README.md -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/Freebase/build_vocab_from_dep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/Freebase/build_vocab_from_dep.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/Freebase/get_2hop_subgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/Freebase/get_2hop_subgraph.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/Freebase/get_seed_set.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/Freebase/get_seed_set.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/Freebase/load_emb_glove.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/Freebase/load_emb_glove.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/Freebase/manual_filter_rel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/Freebase/manual_filter_rel.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/Freebase/map_kb_id.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/Freebase/map_kb_id.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/Freebase/preprocess_step0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/Freebase/preprocess_step0.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/Freebase/preprocess_step1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/Freebase/preprocess_step1.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/Freebase/update_vocab_with_rel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/Freebase/update_vocab_with_rel.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/Freebase/util/deal_cvt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/Freebase/util/deal_cvt.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/Freebase/util/ppr_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/Freebase/util/ppr_util.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/MetaQA/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/MetaQA/README.md -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/MetaQA/build_vocab_from_dep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/MetaQA/build_vocab_from_dep.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/MetaQA/load_emb_glove.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/MetaQA/load_emb_glove.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/MetaQA/map_kb_id.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/MetaQA/map_kb_id.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/MetaQA/ppr_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/MetaQA/ppr_util.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/MetaQA/prepare_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/MetaQA/prepare_data.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/MetaQA/simplify_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/MetaQA/simplify_dataset.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/parse/con_parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/parse/con_parse.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/parse/const_parse.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/parse/const_parse.sh -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/parse/dep_parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/parse/dep_parse.py -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/parse/dependency_parse.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/parse/dependency_parse.sh -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/parse/readme.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/parse/readme.txt -------------------------------------------------------------------------------- /src/nsm_reader/preprocessing/parse/server.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/preprocessing/parse/server.sh -------------------------------------------------------------------------------- /src/nsm_reader/presentation/wsdm-poster.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/presentation/wsdm-poster.pdf -------------------------------------------------------------------------------- /src/nsm_reader/presentation/wsdm_slides_ver2.pptx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/nsm_reader/presentation/wsdm_slides_ver2.pptx -------------------------------------------------------------------------------- /src/preprocessing/load_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/preprocessing/load_dataset.py -------------------------------------------------------------------------------- /src/preprocessing/negative_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/preprocessing/negative_sampling.py -------------------------------------------------------------------------------- /src/preprocessing/negative_sampling_for_unsup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/preprocessing/negative_sampling_for_unsup.py -------------------------------------------------------------------------------- /src/preprocessing/score_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/preprocessing/score_path.py -------------------------------------------------------------------------------- /src/preprocessing/search_to_get_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/preprocessing/search_to_get_path.py -------------------------------------------------------------------------------- /src/retrieve_subgraph/build_relation_set.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retrieve_subgraph/build_relation_set.py -------------------------------------------------------------------------------- /src/retrieve_subgraph/retrieve_subgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retrieve_subgraph/retrieve_subgraph.py -------------------------------------------------------------------------------- /src/retrieve_subgraph/retrieve_subgraph_for_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retrieve_subgraph/retrieve_subgraph_for_finetune.py -------------------------------------------------------------------------------- /src/retrieve_subgraph/retrieve_subgraph_for_graftnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retrieve_subgraph/retrieve_subgraph_for_graftnet.py -------------------------------------------------------------------------------- /src/retrieve_subgraph/retrieve_subgraph_for_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retrieve_subgraph/retrieve_subgraph_for_test.py -------------------------------------------------------------------------------- /src/retrieve_subgraph/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retrieve_subgraph/utils.py -------------------------------------------------------------------------------- /src/retriever/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever/LICENSE -------------------------------------------------------------------------------- /src/retriever/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever/evaluation.py -------------------------------------------------------------------------------- /src/retriever/huggingface_to_simcse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever/huggingface_to_simcse.py -------------------------------------------------------------------------------- /src/retriever/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever/requirements.txt -------------------------------------------------------------------------------- /src/retriever/simcse_to_huggingface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever/simcse_to_huggingface.py -------------------------------------------------------------------------------- /src/retriever/src/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever/src/models.py -------------------------------------------------------------------------------- /src/retriever/src/trainers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever/src/trainers.py -------------------------------------------------------------------------------- /src/retriever/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever/train.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Agent/BaseAgent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Agent/BaseAgent.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Agent/NSMAgent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Agent/NSMAgent.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Agent/TeacherAgent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Agent/TeacherAgent.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Agent/TeacherAgent2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Agent/TeacherAgent2.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Model/backward_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Model/backward_model.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Model/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Model/base_model.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Model/forward_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Model/forward_model.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Model/hybrid_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Model/hybrid_model.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Model/nsm_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Model/nsm_model.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Modules/Instruction/base_instruction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Modules/Instruction/base_instruction.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Modules/Instruction/seq_instruction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Modules/Instruction/seq_instruction.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Modules/Reasoning/base_reasoning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Modules/Reasoning/base_reasoning.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Modules/Reasoning/gnn_backward_reasoning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Modules/Reasoning/gnn_backward_reasoning.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Modules/Reasoning/gnn_reasoning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Modules/Reasoning/gnn_reasoning.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Modules/layer_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Modules/layer_nsm.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/Modules/layers_att.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/Modules/layers_att.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/data/basic_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/data/basic_dataset.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/data/dataset_single.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/data/dataset_single.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/data/dataset_super.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/data/dataset_super.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/data/load_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/data/load_data.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/data/load_data_super.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/data/load_data_super.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/data/read_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/data/read_tree.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/data/util.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/train/evaluate_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/train/evaluate_nsm.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/train/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/train/init.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/train/trainer_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/train/trainer_hybrid.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/train/trainer_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/train/trainer_nsm.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/train/trainer_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/train/trainer_parallel.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/train/trainer_student.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/train/trainer_student.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/util/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/util/config.py -------------------------------------------------------------------------------- /src/retriever_finetune/NSM/util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/NSM/util/utils.py -------------------------------------------------------------------------------- /src/retriever_finetune/build_search_tree_for_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/build_search_tree_for_dataset.py -------------------------------------------------------------------------------- /src/retriever_finetune/build_train_set_from_search_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/build_train_set_from_search_tree.py -------------------------------------------------------------------------------- /src/retriever_finetune/cal_path_score_from_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/cal_path_score_from_reader.py -------------------------------------------------------------------------------- /src/retriever_finetune/m_dict.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/m_dict.json -------------------------------------------------------------------------------- /src/retriever_finetune/main_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/main_nsm.py -------------------------------------------------------------------------------- /src/retriever_finetune/search_tree_list.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/search_tree_list.pkl -------------------------------------------------------------------------------- /src/retriever_finetune/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/retriever_finetune/utils.py -------------------------------------------------------------------------------- /src/run_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/run_preprocess.py -------------------------------------------------------------------------------- /src/run_retrieve_subgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/run_retrieve_subgraph.py -------------------------------------------------------------------------------- /src/run_retriever_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/run_retriever_finetune.py -------------------------------------------------------------------------------- /src/run_retriever_finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/run_retriever_finetune.sh -------------------------------------------------------------------------------- /src/run_train_graftnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/run_train_graftnet.py -------------------------------------------------------------------------------- /src/run_train_graftnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/run_train_graftnet.sh -------------------------------------------------------------------------------- /src/run_train_nsm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/run_train_nsm.py -------------------------------------------------------------------------------- /src/run_train_nsm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/run_train_nsm.sh -------------------------------------------------------------------------------- /src/run_train_retriever.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/run_train_retriever.py -------------------------------------------------------------------------------- /src/run_train_retriever.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/run_train_retriever.sh -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RUCKBReasoning/SubgraphRetrievalKBQA/HEAD/src/utils.py --------------------------------------------------------------------------------