├── .dockerignore ├── .github ├── CONTRIBUTING.md ├── ISSUE_TEMPLATE │ ├── bug_report.md │ └── feature_request.md ├── config │ └── .codespellignore ├── pull_request_template.md ├── scripts │ ├── merge_json_files.py │ ├── poll_slurm_job.sh │ ├── run_download_tgb_job_sbatch.sh │ └── run_perf_job_sbatch.sh └── workflows │ ├── docker.yml │ ├── docs.yml │ ├── integration.yml │ ├── mypy.yml │ ├── performance.yml │ ├── publish.yml │ ├── ruff.yml │ ├── spelling.yml │ └── testing.yml ├── .gitignore ├── .pre-commit-config.yaml ├── .python-version ├── .readthedocs.yaml ├── LICENSE ├── README.md ├── docker ├── Dockerfile.cpu └── Dockerfile.gpu ├── docs ├── api │ ├── batch.md │ ├── constants.md │ ├── data.md │ ├── graph.md │ ├── hooks │ │ ├── hook_manager.md │ │ ├── hooks.md │ │ └── recipe.md │ ├── loader.md │ ├── nn.md │ ├── timedelta.md │ └── util.md ├── architecture.md ├── img │ ├── architecture-dark.svg │ ├── architecture-gray.svg │ ├── architecture-light.svg │ ├── logo.svg │ └── thumbnail.svg ├── index.md ├── quickstart.md └── tutorials │ ├── dgraph_tutorial.md │ ├── hook_tutorial.md │ └── time_delta_tutorial.md ├── examples ├── analytics │ └── dos.py ├── graphproppred │ ├── persistant_forecast.py │ └── tgcn.py ├── linkproppred │ ├── dygformer.py │ ├── edgebank.py │ ├── gclstm.py │ ├── gcn.py │ ├── graphmixer.py │ ├── tgat.py │ ├── tgn.py │ └── tpnet.py └── nodeproppred │ ├── dygformer.py │ ├── gclstm.py │ ├── gcn.py │ ├── persistant_forecast.py │ ├── tgcn.py │ └── tgn.py ├── mkdocs.yml ├── pyproject.toml ├── scripts ├── build_docs.sh ├── download_graph_prop_datasets.sh ├── download_tgb_datasets.sh ├── parse_log_to_json.sh ├── run_perf_tests.sh ├── run_unit_tests.sh ├── trigger_integration_tests.sh ├── trigger_perf_tests.sh └── trigger_workflow.sh ├── test ├── __init__.py ├── integration │ ├── __init__.py │ ├── conftest.py │ ├── test_dygformer.py │ ├── test_edgebank.py │ ├── test_gclstm.py │ ├── test_gcn.py │ ├── test_graphmixer.py │ ├── test_persistant_forecast.py │ ├── test_tgat.py │ ├── test_tgcn.py │ ├── test_tgn.py │ └── test_tpnet.py ├── performance │ ├── __init__.py │ ├── conftest.py │ ├── test_construction.py │ ├── test_discretization.py │ └── test_iteration.py └── unit │ ├── __init__.py │ ├── test_core │ ├── test_dgraph.py │ ├── test_storage_impl.py │ └── test_timedelta.py │ ├── test_data │ ├── test_data.py │ ├── test_dataloader.py │ └── test_split.py │ ├── test_hooks │ ├── test_deduplication_hook.py │ ├── test_device_transfer_hook.py │ ├── test_hook_manager.py │ ├── test_negative_edge_sampler_hook.py │ ├── test_neighbor_sampler_hook.py │ ├── test_pin_memory_hook.py │ ├── test_recency_nbr_hook.py │ ├── test_recipe.py │ ├── test_seen_nodes_track_hook.py │ └── test_tgb_negative_sampling_hook.py │ ├── test_nn │ ├── conftest.py │ ├── test_dygformer.py │ ├── test_edgebank.py │ ├── test_gclstm.py │ ├── test_graphdecoder.py │ ├── test_linkdecoder.py │ ├── test_mlp_mixer.py │ ├── test_nodedecoder.py │ ├── test_temporal_attention.py │ ├── test_tgcn.py │ ├── test_time2vec.py │ └── test_tpnet.py │ └── test_util │ ├── test_logging.py │ └── test_seed.py ├── tgm ├── __init__.py ├── constants.py ├── core │ ├── __init__.py │ ├── _storage │ │ ├── __init__.py │ │ ├── backends │ │ │ ├── __init__.py │ │ │ └── array_backend.py │ │ └── base.py │ ├── batch.py │ ├── graph.py │ └── timedelta.py ├── data │ ├── __init__.py │ ├── dg_data.py │ ├── loader.py │ └── split.py ├── exceptions.py ├── hooks │ ├── __init__.py │ ├── base.py │ ├── dedup.py │ ├── device.py │ ├── hook_manager.py │ ├── negatives.py │ ├── neighbors.py │ ├── node_tracks.py │ └── recipe.py ├── nn │ ├── __init__.py │ ├── decoder │ │ ├── __init__.py │ │ ├── graphproppred.py │ │ ├── linkproppred.py │ │ └── nodeproppred.py │ ├── encoder │ │ ├── __init__.py │ │ ├── dygformer.py │ │ ├── gclstm.py │ │ ├── tgcn.py │ │ └── tpnet.py │ └── modules │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── edgebank.py │ │ ├── mlp_mixer.py │ │ └── time_encoding.py └── util │ ├── __init__.py │ ├── logging.py │ └── seed.py ├── tools ├── log_parser.py └── profiling.py └── uv.lock /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.dockerignore -------------------------------------------------------------------------------- /.github/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/CONTRIBUTING.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/config/.codespellignore: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /.github/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/pull_request_template.md -------------------------------------------------------------------------------- /.github/scripts/merge_json_files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/scripts/merge_json_files.py -------------------------------------------------------------------------------- /.github/scripts/poll_slurm_job.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/scripts/poll_slurm_job.sh -------------------------------------------------------------------------------- /.github/scripts/run_download_tgb_job_sbatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/scripts/run_download_tgb_job_sbatch.sh -------------------------------------------------------------------------------- /.github/scripts/run_perf_job_sbatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/scripts/run_perf_job_sbatch.sh -------------------------------------------------------------------------------- /.github/workflows/docker.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/workflows/docker.yml -------------------------------------------------------------------------------- /.github/workflows/docs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/workflows/docs.yml -------------------------------------------------------------------------------- /.github/workflows/integration.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/workflows/integration.yml -------------------------------------------------------------------------------- /.github/workflows/mypy.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/workflows/mypy.yml -------------------------------------------------------------------------------- /.github/workflows/performance.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/workflows/performance.yml -------------------------------------------------------------------------------- /.github/workflows/publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/workflows/publish.yml -------------------------------------------------------------------------------- /.github/workflows/ruff.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/workflows/ruff.yml -------------------------------------------------------------------------------- /.github/workflows/spelling.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/workflows/spelling.yml -------------------------------------------------------------------------------- /.github/workflows/testing.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.github/workflows/testing.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.python-version: -------------------------------------------------------------------------------- 1 | 3.10 2 | -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/README.md -------------------------------------------------------------------------------- /docker/Dockerfile.cpu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docker/Dockerfile.cpu -------------------------------------------------------------------------------- /docker/Dockerfile.gpu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docker/Dockerfile.gpu -------------------------------------------------------------------------------- /docs/api/batch.md: -------------------------------------------------------------------------------- 1 | ::: tgm.DGBatch 2 | -------------------------------------------------------------------------------- /docs/api/constants.md: -------------------------------------------------------------------------------- 1 | ::: tgm.constants 2 | -------------------------------------------------------------------------------- /docs/api/data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/api/data.md -------------------------------------------------------------------------------- /docs/api/graph.md: -------------------------------------------------------------------------------- 1 | ::: tgm.DGraph 2 | -------------------------------------------------------------------------------- /docs/api/hooks/hook_manager.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/api/hooks/hook_manager.md -------------------------------------------------------------------------------- /docs/api/hooks/hooks.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/api/hooks/hooks.md -------------------------------------------------------------------------------- /docs/api/hooks/recipe.md: -------------------------------------------------------------------------------- 1 | ::: tgm.hooks.recipe 2 | -------------------------------------------------------------------------------- /docs/api/loader.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/api/loader.md -------------------------------------------------------------------------------- /docs/api/nn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/api/nn.md -------------------------------------------------------------------------------- /docs/api/timedelta.md: -------------------------------------------------------------------------------- 1 | ::: tgm.core.timedelta 2 | -------------------------------------------------------------------------------- /docs/api/util.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/api/util.md -------------------------------------------------------------------------------- /docs/architecture.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/architecture.md -------------------------------------------------------------------------------- /docs/img/architecture-dark.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/img/architecture-dark.svg -------------------------------------------------------------------------------- /docs/img/architecture-gray.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/img/architecture-gray.svg -------------------------------------------------------------------------------- /docs/img/architecture-light.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/img/architecture-light.svg -------------------------------------------------------------------------------- /docs/img/logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/img/logo.svg -------------------------------------------------------------------------------- /docs/img/thumbnail.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/img/thumbnail.svg -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/quickstart.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/quickstart.md -------------------------------------------------------------------------------- /docs/tutorials/dgraph_tutorial.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/tutorials/dgraph_tutorial.md -------------------------------------------------------------------------------- /docs/tutorials/hook_tutorial.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/tutorials/hook_tutorial.md -------------------------------------------------------------------------------- /docs/tutorials/time_delta_tutorial.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/docs/tutorials/time_delta_tutorial.md -------------------------------------------------------------------------------- /examples/analytics/dos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/analytics/dos.py -------------------------------------------------------------------------------- /examples/graphproppred/persistant_forecast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/graphproppred/persistant_forecast.py -------------------------------------------------------------------------------- /examples/graphproppred/tgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/graphproppred/tgcn.py -------------------------------------------------------------------------------- /examples/linkproppred/dygformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/linkproppred/dygformer.py -------------------------------------------------------------------------------- /examples/linkproppred/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/linkproppred/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/gclstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/linkproppred/gclstm.py -------------------------------------------------------------------------------- /examples/linkproppred/gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/linkproppred/gcn.py -------------------------------------------------------------------------------- /examples/linkproppred/graphmixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/linkproppred/graphmixer.py -------------------------------------------------------------------------------- /examples/linkproppred/tgat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/linkproppred/tgat.py -------------------------------------------------------------------------------- /examples/linkproppred/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/linkproppred/tgn.py -------------------------------------------------------------------------------- /examples/linkproppred/tpnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/linkproppred/tpnet.py -------------------------------------------------------------------------------- /examples/nodeproppred/dygformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/nodeproppred/dygformer.py -------------------------------------------------------------------------------- /examples/nodeproppred/gclstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/nodeproppred/gclstm.py -------------------------------------------------------------------------------- /examples/nodeproppred/gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/nodeproppred/gcn.py -------------------------------------------------------------------------------- /examples/nodeproppred/persistant_forecast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/nodeproppred/persistant_forecast.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/nodeproppred/tgcn.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/examples/nodeproppred/tgn.py -------------------------------------------------------------------------------- /mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/mkdocs.yml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/pyproject.toml -------------------------------------------------------------------------------- /scripts/build_docs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/scripts/build_docs.sh -------------------------------------------------------------------------------- /scripts/download_graph_prop_datasets.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/scripts/download_graph_prop_datasets.sh -------------------------------------------------------------------------------- /scripts/download_tgb_datasets.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/scripts/download_tgb_datasets.sh -------------------------------------------------------------------------------- /scripts/parse_log_to_json.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/scripts/parse_log_to_json.sh -------------------------------------------------------------------------------- /scripts/run_perf_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/scripts/run_perf_tests.sh -------------------------------------------------------------------------------- /scripts/run_unit_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/scripts/run_unit_tests.sh -------------------------------------------------------------------------------- /scripts/trigger_integration_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/scripts/trigger_integration_tests.sh -------------------------------------------------------------------------------- /scripts/trigger_perf_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/scripts/trigger_perf_tests.sh -------------------------------------------------------------------------------- /scripts/trigger_workflow.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/scripts/trigger_workflow.sh -------------------------------------------------------------------------------- /test/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/integration/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/integration/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/integration/conftest.py -------------------------------------------------------------------------------- /test/integration/test_dygformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/integration/test_dygformer.py -------------------------------------------------------------------------------- /test/integration/test_edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/integration/test_edgebank.py -------------------------------------------------------------------------------- /test/integration/test_gclstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/integration/test_gclstm.py -------------------------------------------------------------------------------- /test/integration/test_gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/integration/test_gcn.py -------------------------------------------------------------------------------- /test/integration/test_graphmixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/integration/test_graphmixer.py -------------------------------------------------------------------------------- /test/integration/test_persistant_forecast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/integration/test_persistant_forecast.py -------------------------------------------------------------------------------- /test/integration/test_tgat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/integration/test_tgat.py -------------------------------------------------------------------------------- /test/integration/test_tgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/integration/test_tgcn.py -------------------------------------------------------------------------------- /test/integration/test_tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/integration/test_tgn.py -------------------------------------------------------------------------------- /test/integration/test_tpnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/integration/test_tpnet.py -------------------------------------------------------------------------------- /test/performance/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/performance/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/performance/conftest.py -------------------------------------------------------------------------------- /test/performance/test_construction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/performance/test_construction.py -------------------------------------------------------------------------------- /test/performance/test_discretization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/performance/test_discretization.py -------------------------------------------------------------------------------- /test/performance/test_iteration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/performance/test_iteration.py -------------------------------------------------------------------------------- /test/unit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/unit/test_core/test_dgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_core/test_dgraph.py -------------------------------------------------------------------------------- /test/unit/test_core/test_storage_impl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_core/test_storage_impl.py -------------------------------------------------------------------------------- /test/unit/test_core/test_timedelta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_core/test_timedelta.py -------------------------------------------------------------------------------- /test/unit/test_data/test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_data/test_data.py -------------------------------------------------------------------------------- /test/unit/test_data/test_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_data/test_dataloader.py -------------------------------------------------------------------------------- /test/unit/test_data/test_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_data/test_split.py -------------------------------------------------------------------------------- /test/unit/test_hooks/test_deduplication_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_hooks/test_deduplication_hook.py -------------------------------------------------------------------------------- /test/unit/test_hooks/test_device_transfer_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_hooks/test_device_transfer_hook.py -------------------------------------------------------------------------------- /test/unit/test_hooks/test_hook_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_hooks/test_hook_manager.py -------------------------------------------------------------------------------- /test/unit/test_hooks/test_negative_edge_sampler_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_hooks/test_negative_edge_sampler_hook.py -------------------------------------------------------------------------------- /test/unit/test_hooks/test_neighbor_sampler_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_hooks/test_neighbor_sampler_hook.py -------------------------------------------------------------------------------- /test/unit/test_hooks/test_pin_memory_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_hooks/test_pin_memory_hook.py -------------------------------------------------------------------------------- /test/unit/test_hooks/test_recency_nbr_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_hooks/test_recency_nbr_hook.py -------------------------------------------------------------------------------- /test/unit/test_hooks/test_recipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_hooks/test_recipe.py -------------------------------------------------------------------------------- /test/unit/test_hooks/test_seen_nodes_track_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_hooks/test_seen_nodes_track_hook.py -------------------------------------------------------------------------------- /test/unit/test_hooks/test_tgb_negative_sampling_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_hooks/test_tgb_negative_sampling_hook.py -------------------------------------------------------------------------------- /test/unit/test_nn/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_nn/conftest.py -------------------------------------------------------------------------------- /test/unit/test_nn/test_dygformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_nn/test_dygformer.py -------------------------------------------------------------------------------- /test/unit/test_nn/test_edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_nn/test_edgebank.py -------------------------------------------------------------------------------- /test/unit/test_nn/test_gclstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_nn/test_gclstm.py -------------------------------------------------------------------------------- /test/unit/test_nn/test_graphdecoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_nn/test_graphdecoder.py -------------------------------------------------------------------------------- /test/unit/test_nn/test_linkdecoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_nn/test_linkdecoder.py -------------------------------------------------------------------------------- /test/unit/test_nn/test_mlp_mixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_nn/test_mlp_mixer.py -------------------------------------------------------------------------------- /test/unit/test_nn/test_nodedecoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_nn/test_nodedecoder.py -------------------------------------------------------------------------------- /test/unit/test_nn/test_temporal_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_nn/test_temporal_attention.py -------------------------------------------------------------------------------- /test/unit/test_nn/test_tgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_nn/test_tgcn.py -------------------------------------------------------------------------------- /test/unit/test_nn/test_time2vec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_nn/test_time2vec.py -------------------------------------------------------------------------------- /test/unit/test_nn/test_tpnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_nn/test_tpnet.py -------------------------------------------------------------------------------- /test/unit/test_util/test_logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_util/test_logging.py -------------------------------------------------------------------------------- /test/unit/test_util/test_seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/test/unit/test_util/test_seed.py -------------------------------------------------------------------------------- /tgm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/__init__.py -------------------------------------------------------------------------------- /tgm/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/constants.py -------------------------------------------------------------------------------- /tgm/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/core/__init__.py -------------------------------------------------------------------------------- /tgm/core/_storage/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/core/_storage/__init__.py -------------------------------------------------------------------------------- /tgm/core/_storage/backends/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/core/_storage/backends/__init__.py -------------------------------------------------------------------------------- /tgm/core/_storage/backends/array_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/core/_storage/backends/array_backend.py -------------------------------------------------------------------------------- /tgm/core/_storage/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/core/_storage/base.py -------------------------------------------------------------------------------- /tgm/core/batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/core/batch.py -------------------------------------------------------------------------------- /tgm/core/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/core/graph.py -------------------------------------------------------------------------------- /tgm/core/timedelta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/core/timedelta.py -------------------------------------------------------------------------------- /tgm/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/data/__init__.py -------------------------------------------------------------------------------- /tgm/data/dg_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/data/dg_data.py -------------------------------------------------------------------------------- /tgm/data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/data/loader.py -------------------------------------------------------------------------------- /tgm/data/split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/data/split.py -------------------------------------------------------------------------------- /tgm/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/exceptions.py -------------------------------------------------------------------------------- /tgm/hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/hooks/__init__.py -------------------------------------------------------------------------------- /tgm/hooks/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/hooks/base.py -------------------------------------------------------------------------------- /tgm/hooks/dedup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/hooks/dedup.py -------------------------------------------------------------------------------- /tgm/hooks/device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/hooks/device.py -------------------------------------------------------------------------------- /tgm/hooks/hook_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/hooks/hook_manager.py -------------------------------------------------------------------------------- /tgm/hooks/negatives.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/hooks/negatives.py -------------------------------------------------------------------------------- /tgm/hooks/neighbors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/hooks/neighbors.py -------------------------------------------------------------------------------- /tgm/hooks/node_tracks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/hooks/node_tracks.py -------------------------------------------------------------------------------- /tgm/hooks/recipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/hooks/recipe.py -------------------------------------------------------------------------------- /tgm/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/nn/__init__.py -------------------------------------------------------------------------------- /tgm/nn/decoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/nn/decoder/__init__.py -------------------------------------------------------------------------------- /tgm/nn/decoder/graphproppred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/nn/decoder/graphproppred.py -------------------------------------------------------------------------------- /tgm/nn/decoder/linkproppred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/nn/decoder/linkproppred.py -------------------------------------------------------------------------------- /tgm/nn/decoder/nodeproppred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/nn/decoder/nodeproppred.py -------------------------------------------------------------------------------- /tgm/nn/encoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/nn/encoder/__init__.py -------------------------------------------------------------------------------- /tgm/nn/encoder/dygformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/nn/encoder/dygformer.py -------------------------------------------------------------------------------- /tgm/nn/encoder/gclstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/nn/encoder/gclstm.py -------------------------------------------------------------------------------- /tgm/nn/encoder/tgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/nn/encoder/tgcn.py -------------------------------------------------------------------------------- /tgm/nn/encoder/tpnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/nn/encoder/tpnet.py -------------------------------------------------------------------------------- /tgm/nn/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/nn/modules/__init__.py -------------------------------------------------------------------------------- /tgm/nn/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/nn/modules/attention.py -------------------------------------------------------------------------------- /tgm/nn/modules/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/nn/modules/edgebank.py -------------------------------------------------------------------------------- /tgm/nn/modules/mlp_mixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/nn/modules/mlp_mixer.py -------------------------------------------------------------------------------- /tgm/nn/modules/time_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/nn/modules/time_encoding.py -------------------------------------------------------------------------------- /tgm/util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tgm/util/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/util/logging.py -------------------------------------------------------------------------------- /tgm/util/seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tgm/util/seed.py -------------------------------------------------------------------------------- /tools/log_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tools/log_parser.py -------------------------------------------------------------------------------- /tools/profiling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/tools/profiling.py -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tgm-team/tgm/HEAD/uv.lock --------------------------------------------------------------------------------