├── .github ├── actions │ └── setup │ │ └── action.yml └── workflows │ ├── linting.yml │ └── testing.yml ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── benchmark ├── relbench_link_prediction_benchmark.py ├── relbench_multi_vae.py └── tgt_ijcai_benchmark.py ├── contextgnn ├── __init__.py ├── data │ ├── __init__.py │ └── ijcai_contest.py ├── nn │ ├── __init__.py │ ├── encoder.py │ ├── models │ │ ├── __init__.py │ │ ├── contextgnn.py │ │ ├── graphsage.py │ │ ├── idgnn.py │ │ ├── rhsembeddinggnn.py │ │ └── shallowrhsgnn.py │ └── rhs_embedding.py └── utils │ ├── __init__.py │ ├── metrics.py │ └── text_embedder.py ├── examples ├── contextgnn_sample_softmax.py ├── ijcai_example.py ├── light_gcn.py ├── ngcf.py └── relbench_example.py ├── pyproject.toml └── test └── nn ├── test_encoder.py ├── test_model.py └── test_rhs_embedding.py /.github/actions/setup/action.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/.github/actions/setup/action.yml -------------------------------------------------------------------------------- /.github/workflows/linting.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/.github/workflows/linting.yml -------------------------------------------------------------------------------- /.github/workflows/testing.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/.github/workflows/testing.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/README.md -------------------------------------------------------------------------------- /benchmark/relbench_link_prediction_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/benchmark/relbench_link_prediction_benchmark.py -------------------------------------------------------------------------------- /benchmark/relbench_multi_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/benchmark/relbench_multi_vae.py -------------------------------------------------------------------------------- /benchmark/tgt_ijcai_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/benchmark/tgt_ijcai_benchmark.py -------------------------------------------------------------------------------- /contextgnn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /contextgnn/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/contextgnn/data/__init__.py -------------------------------------------------------------------------------- /contextgnn/data/ijcai_contest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/contextgnn/data/ijcai_contest.py -------------------------------------------------------------------------------- /contextgnn/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/contextgnn/nn/__init__.py -------------------------------------------------------------------------------- /contextgnn/nn/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/contextgnn/nn/encoder.py -------------------------------------------------------------------------------- /contextgnn/nn/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/contextgnn/nn/models/__init__.py -------------------------------------------------------------------------------- /contextgnn/nn/models/contextgnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/contextgnn/nn/models/contextgnn.py -------------------------------------------------------------------------------- /contextgnn/nn/models/graphsage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/contextgnn/nn/models/graphsage.py -------------------------------------------------------------------------------- /contextgnn/nn/models/idgnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/contextgnn/nn/models/idgnn.py -------------------------------------------------------------------------------- /contextgnn/nn/models/rhsembeddinggnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/contextgnn/nn/models/rhsembeddinggnn.py -------------------------------------------------------------------------------- /contextgnn/nn/models/shallowrhsgnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/contextgnn/nn/models/shallowrhsgnn.py -------------------------------------------------------------------------------- /contextgnn/nn/rhs_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/contextgnn/nn/rhs_embedding.py -------------------------------------------------------------------------------- /contextgnn/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/contextgnn/utils/__init__.py -------------------------------------------------------------------------------- /contextgnn/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/contextgnn/utils/metrics.py -------------------------------------------------------------------------------- /contextgnn/utils/text_embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/contextgnn/utils/text_embedder.py -------------------------------------------------------------------------------- /examples/contextgnn_sample_softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/examples/contextgnn_sample_softmax.py -------------------------------------------------------------------------------- /examples/ijcai_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/examples/ijcai_example.py -------------------------------------------------------------------------------- /examples/light_gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/examples/light_gcn.py -------------------------------------------------------------------------------- /examples/ngcf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/examples/ngcf.py -------------------------------------------------------------------------------- /examples/relbench_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/examples/relbench_example.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/pyproject.toml -------------------------------------------------------------------------------- /test/nn/test_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/test/nn/test_encoder.py -------------------------------------------------------------------------------- /test/nn/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/test/nn/test_model.py -------------------------------------------------------------------------------- /test/nn/test_rhs_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumo-ai/ContextGNN/HEAD/test/nn/test_rhs_embedding.py --------------------------------------------------------------------------------