├── .env.template ├── .gitignore ├── LICENSE ├── README.md ├── STaRKQADataset.py ├── STaRKQADatasetGDS.py ├── STaRKQAVectorSearchDataset.py ├── architecture.png ├── compute_metrics.py ├── compute_pcst.py ├── configs ├── algo_config_v0.yaml └── retrieval_config_v0.yaml ├── data-loading ├── emb │ └── prime │ │ └── text-embedding-ada-002 │ │ ├── doc │ │ ├── reltype_emb_dict.pt │ │ └── triplet_sentence_emb_dict.pt │ │ └── query │ │ └── query_emb_dict.pt ├── emb_download.py ├── load_data.py └── stark_prime_neo4j_loading.ipynb ├── db.env ├── eval_pcst_ordering.ipynb ├── finalmetric.png ├── lecture_corpus_txt2kg_nvidia ├── README.md ├── lecture_corpus.env ├── lecture_corpus_kg_ex1.png ├── lecture_corpus_kg_ex2.png ├── load_txt2kg_to_neo4j.py ├── requirements.txt ├── txt2kg_rag.py └── txt2kg_rag_utils.py ├── main.py ├── plot_pr.py ├── plot_results.py ├── plotpr.png ├── requirements.txt ├── retrieve-prime-subgraphs.ipynb ├── setup_db_env.py └── train.py /.env.template: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/.env.template -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/README.md -------------------------------------------------------------------------------- /STaRKQADataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/STaRKQADataset.py -------------------------------------------------------------------------------- /STaRKQADatasetGDS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/STaRKQADatasetGDS.py -------------------------------------------------------------------------------- /STaRKQAVectorSearchDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/STaRKQAVectorSearchDataset.py -------------------------------------------------------------------------------- /architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/architecture.png -------------------------------------------------------------------------------- /compute_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/compute_metrics.py -------------------------------------------------------------------------------- /compute_pcst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/compute_pcst.py -------------------------------------------------------------------------------- /configs/algo_config_v0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/configs/algo_config_v0.yaml -------------------------------------------------------------------------------- /configs/retrieval_config_v0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/configs/retrieval_config_v0.yaml -------------------------------------------------------------------------------- /data-loading/emb/prime/text-embedding-ada-002/doc/reltype_emb_dict.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/data-loading/emb/prime/text-embedding-ada-002/doc/reltype_emb_dict.pt -------------------------------------------------------------------------------- /data-loading/emb/prime/text-embedding-ada-002/doc/triplet_sentence_emb_dict.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/data-loading/emb/prime/text-embedding-ada-002/doc/triplet_sentence_emb_dict.pt -------------------------------------------------------------------------------- /data-loading/emb/prime/text-embedding-ada-002/query/query_emb_dict.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/data-loading/emb/prime/text-embedding-ada-002/query/query_emb_dict.pt -------------------------------------------------------------------------------- /data-loading/emb_download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/data-loading/emb_download.py -------------------------------------------------------------------------------- /data-loading/load_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/data-loading/load_data.py -------------------------------------------------------------------------------- /data-loading/stark_prime_neo4j_loading.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/data-loading/stark_prime_neo4j_loading.ipynb -------------------------------------------------------------------------------- /db.env: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/db.env -------------------------------------------------------------------------------- /eval_pcst_ordering.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/eval_pcst_ordering.ipynb -------------------------------------------------------------------------------- /finalmetric.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/finalmetric.png -------------------------------------------------------------------------------- /lecture_corpus_txt2kg_nvidia/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/lecture_corpus_txt2kg_nvidia/README.md -------------------------------------------------------------------------------- /lecture_corpus_txt2kg_nvidia/lecture_corpus.env: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/lecture_corpus_txt2kg_nvidia/lecture_corpus.env -------------------------------------------------------------------------------- /lecture_corpus_txt2kg_nvidia/lecture_corpus_kg_ex1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/lecture_corpus_txt2kg_nvidia/lecture_corpus_kg_ex1.png -------------------------------------------------------------------------------- /lecture_corpus_txt2kg_nvidia/lecture_corpus_kg_ex2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/lecture_corpus_txt2kg_nvidia/lecture_corpus_kg_ex2.png -------------------------------------------------------------------------------- /lecture_corpus_txt2kg_nvidia/load_txt2kg_to_neo4j.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/lecture_corpus_txt2kg_nvidia/load_txt2kg_to_neo4j.py -------------------------------------------------------------------------------- /lecture_corpus_txt2kg_nvidia/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/lecture_corpus_txt2kg_nvidia/requirements.txt -------------------------------------------------------------------------------- /lecture_corpus_txt2kg_nvidia/txt2kg_rag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/lecture_corpus_txt2kg_nvidia/txt2kg_rag.py -------------------------------------------------------------------------------- /lecture_corpus_txt2kg_nvidia/txt2kg_rag_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/lecture_corpus_txt2kg_nvidia/txt2kg_rag_utils.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/main.py -------------------------------------------------------------------------------- /plot_pr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/plot_pr.py -------------------------------------------------------------------------------- /plot_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/plot_results.py -------------------------------------------------------------------------------- /plotpr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/plotpr.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/requirements.txt -------------------------------------------------------------------------------- /retrieve-prime-subgraphs.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/retrieve-prime-subgraphs.ipynb -------------------------------------------------------------------------------- /setup_db_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/setup_db_env.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/neo4j-product-examples/neo4j-gnn-llm-example/HEAD/train.py --------------------------------------------------------------------------------