├── .devcontainer ├── .gitignore ├── Dockerfile └── devcontainer.json ├── .github └── workflows │ ├── mkdocs.yaml │ └── pypi.yaml ├── .gitignore ├── LICENSE ├── README.md ├── docs ├── about.md ├── api │ ├── tgb.linkproppred.md │ ├── tgb.nodeproppred.md │ └── tgb.utils.md ├── assets │ └── logo.png ├── index.md └── tutorials │ ├── Edge_data_numpy.ipynb │ ├── Edge_data_pyg.ipynb │ └── Node_label_tutorial.ipynb ├── examples ├── linkproppred │ ├── tgbl-coin │ │ ├── dyrep.py │ │ ├── edgebank.py │ │ └── tgn.py │ ├── tgbl-comment │ │ ├── dyrep.py │ │ ├── edgebank.py │ │ └── tgn.py │ ├── tgbl-enron │ │ └── edgebank.py │ ├── tgbl-flight │ │ ├── dyrep.py │ │ ├── edgebank.py │ │ └── tgn.py │ ├── tgbl-lastfm │ │ ├── edgebank.py │ │ └── tgn.py │ ├── tgbl-review │ │ ├── dyrep.py │ │ ├── edgebank.py │ │ └── tgn.py │ ├── tgbl-subreddit │ │ ├── edgebank.py │ │ └── tgn.py │ ├── tgbl-uci │ │ └── edgebank.py │ ├── tgbl-wiki │ │ ├── dyrep.py │ │ ├── edgebank.py │ │ └── tgn.py │ ├── thgl-forum │ │ ├── edgebank.py │ │ ├── recurrencybaseline.py │ │ ├── sthn.py │ │ └── tgn.py │ ├── thgl-github │ │ ├── edgebank.py │ │ ├── recurrencybaseline.py │ │ ├── run_seeds.sh │ │ ├── sthn.py │ │ └── tgn.py │ ├── thgl-myket │ │ ├── edgebank.py │ │ ├── recurrencybaseline.py │ │ ├── sthn.py │ │ └── tgn.py │ ├── thgl-software │ │ ├── STHN_README.md │ │ ├── edgebank.py │ │ ├── recurrencybaseline.py │ │ ├── sthn.py │ │ └── tgn.py │ ├── tkgl-icews │ │ ├── cen.py │ │ ├── edgebank.py │ │ ├── recurrencybaseline.py │ │ ├── regcn.py │ │ ├── timetraveler.py │ │ ├── tkgl-icews_example.py │ │ └── tlogic.py │ ├── tkgl-polecat │ │ ├── cen.py │ │ ├── edgebank.py │ │ ├── example.py │ │ ├── recurrencybaseline.py │ │ ├── regcn.py │ │ ├── timetraveler.py │ │ ├── tkgl-polecat_example.py │ │ └── tlogic.py │ ├── tkgl-smallpedia │ │ ├── cen.py │ │ ├── edgebank.py │ │ ├── recurrencybaseline.py │ │ ├── regcn.py │ │ ├── timetraveler.py │ │ ├── tkgl-smallpedia_example.py │ │ └── tlogic.py │ ├── tkgl-wikidata │ │ ├── edgebank.py │ │ ├── recurrencybaseline.py │ │ ├── regcn.py │ │ ├── tkgl-wikidata_example.py │ │ └── tlogic.py │ └── tkgl-yago │ │ ├── cen.py │ │ ├── edgebank.py │ │ ├── recurrencybaseline.py │ │ ├── regcn.py │ │ ├── timetraveler.py │ │ ├── tkgl-yago_example.py │ │ └── tlogic.py └── nodeproppred │ ├── tgbn-genre │ ├── dyrep.py │ ├── moving_average.py │ ├── persistant_forecast.py │ └── tgn.py │ ├── tgbn-reddit │ ├── dyrep.py │ ├── moving_average.py │ ├── persistant_forecast.py │ └── tgn.py │ ├── tgbn-token │ ├── dyrep.py │ ├── moving_average.py │ ├── persistant_forecast.py │ └── tgn.py │ └── tgbn-trade │ ├── count_new_nodes.py │ ├── dyrep.py │ ├── moving_average.py │ ├── persistant_forecast.py │ └── tgn.py ├── imgs ├── logo.png ├── pipeline.PNG ├── pipeline.png └── stats.png ├── mkdocs.yml ├── modules ├── decoder.py ├── early_stopping.py ├── edgebank_predictor.py ├── emb_module.py ├── heuristics.py ├── memory_module.py ├── msg_agg.py ├── msg_func.py ├── neighbor_loader.py ├── nodebank.py ├── recurrencybaseline_predictor.py ├── rgcn_layers.py ├── rgcn_model.py ├── rrgcn.py ├── sampler_core.cpp ├── sthn.py ├── sthn_sampler_setup.py ├── time_enc.py ├── timetraveler_agent.py ├── timetraveler_dirichlet.py ├── timetraveler_environment.py ├── timetraveler_episode.py ├── timetraveler_policygradient.py ├── timetraveler_trainertester.py ├── tkg_utils.py ├── tkg_utils_dgl.py ├── tlogic_apply_modules.py └── tlogic_learn_modules.py ├── pipeline.png ├── pyproject.toml ├── requirements-dev.txt ├── requirements.txt ├── run.sh ├── scripts ├── env.sh ├── mila.sh ├── mila_install.sh └── run.sh ├── setup.py ├── stats_figures ├── README.md ├── compute_dataset_stats.py ├── compute_relation_dataset_stats.py ├── compute_relation_results_df.py ├── create_edges_figures.py ├── create_relation_figures.py └── dataset_utils.py └── tgb ├── datasets ├── ICEWS14 │ ├── ent2word.py │ └── icews14.py ├── dataset_scripts │ ├── MAG │ │ ├── mag.py │ │ └── old │ │ │ └── plot_stats.py │ ├── dgraph.py │ ├── dgraph_Readme.md │ ├── process_arxiv.py │ ├── process_github.py │ ├── tgbl-coin.py │ ├── tgbl-coin_neg_generator.py │ ├── tgbl-comment.py │ ├── tgbl-comment_neg_generator.py │ ├── tgbl-flight.py │ ├── tgbl-flight_neg_generator.py │ ├── tgbl-review.py │ ├── tgbl-review_neg_generator.py │ ├── tgbl-wiki_neg_generator.py │ ├── tgbn-genre.py │ ├── tgbn-reddit.py │ ├── tgbn-token.py │ └── tgbn-trade.py ├── tgbl_enron │ ├── tgbl-enron_neg_generator.py │ └── tgbl_enron.py ├── tgbl_lastfm │ └── tgbl-lastfm_neg_generator.py ├── tgbl_subreddit │ └── tgbl-subreddit_neg_generator.py ├── tgbl_uci │ ├── tgbl-uci_neg_generator.py │ └── tgbl_uci.py ├── thgl_forum │ ├── merge_files.py │ ├── num_edges_discretized_thgl-forum.png │ ├── thgl-forum.py │ └── thgl_forum_ns_gen.py ├── thgl_github │ ├── 2024_01 │ │ └── github_extract.py │ ├── 2024_02 │ │ └── github_extract.py │ ├── 2024_03 │ │ └── github_extract.py │ ├── extract_subset.py │ ├── thgl_github.py │ └── thgl_github_ns_gen.py ├── thgl_myket │ ├── thgl_myket.py │ └── thgl_myket_ns_gen.py ├── thgl_software │ ├── thgl_software.py │ └── thgl_software_ns_gen.py ├── tkgl_icews │ ├── tkgl_icews.py │ └── tkgl_icews_ns_gen.py ├── tkgl_polecat │ ├── tkgl_polecat.py │ └── tkgl_polecat_ns_gen.py ├── tkgl_smallpedia │ ├── smallpedia_remove_conflict.py │ └── tkgl_smallpedia_ns_gen.py ├── tkgl_wikidata │ ├── extract.sh │ ├── time_edges │ │ └── tkgl-wikidata_extract.py │ ├── tkgl-wikidata.py │ ├── tkgl_wikidata_mining.py │ ├── tkgl_wikidata_ns_gen.py │ └── wikidata_remove_conflict.py └── tkgl_yago │ ├── tkgl_yago.py │ └── tkgl_yago_ns_gen.py ├── linkproppred ├── dataset.py ├── dataset_pyg.py ├── evaluate.py ├── negative_generator.py ├── negative_sampler.py ├── thg_negative_generator.py ├── thg_negative_sampler.py ├── tkg_negative_generator.py └── tkg_negative_sampler.py ├── nodeproppred ├── dataset.py ├── dataset_pyg.py └── evaluate.py └── utils ├── dataset_stats.py ├── info.py ├── pre_process.py ├── stats.py └── utils.py /.devcontainer/.gitignore: -------------------------------------------------------------------------------- 1 | !devcontainer.json -------------------------------------------------------------------------------- /.devcontainer/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/.devcontainer/Dockerfile -------------------------------------------------------------------------------- /.devcontainer/devcontainer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/.devcontainer/devcontainer.json -------------------------------------------------------------------------------- /.github/workflows/mkdocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/.github/workflows/mkdocs.yaml -------------------------------------------------------------------------------- /.github/workflows/pypi.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/.github/workflows/pypi.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/README.md -------------------------------------------------------------------------------- /docs/about.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/docs/about.md -------------------------------------------------------------------------------- /docs/api/tgb.linkproppred.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/docs/api/tgb.linkproppred.md -------------------------------------------------------------------------------- /docs/api/tgb.nodeproppred.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/docs/api/tgb.nodeproppred.md -------------------------------------------------------------------------------- /docs/api/tgb.utils.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/docs/api/tgb.utils.md -------------------------------------------------------------------------------- /docs/assets/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/docs/assets/logo.png -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/tutorials/Edge_data_numpy.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/docs/tutorials/Edge_data_numpy.ipynb -------------------------------------------------------------------------------- /docs/tutorials/Edge_data_pyg.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/docs/tutorials/Edge_data_pyg.ipynb -------------------------------------------------------------------------------- /docs/tutorials/Node_label_tutorial.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/docs/tutorials/Node_label_tutorial.ipynb -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-coin/dyrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-coin/dyrep.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-coin/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-coin/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-coin/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-coin/tgn.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-comment/dyrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-comment/dyrep.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-comment/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-comment/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-comment/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-comment/tgn.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-enron/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-enron/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-flight/dyrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-flight/dyrep.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-flight/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-flight/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-flight/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-flight/tgn.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-lastfm/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-lastfm/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-lastfm/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-lastfm/tgn.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-review/dyrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-review/dyrep.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-review/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-review/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-review/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-review/tgn.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-subreddit/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-subreddit/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-subreddit/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-subreddit/tgn.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-uci/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-uci/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-wiki/dyrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-wiki/dyrep.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-wiki/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-wiki/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/tgbl-wiki/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tgbl-wiki/tgn.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-forum/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-forum/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-forum/recurrencybaseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-forum/recurrencybaseline.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-forum/sthn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-forum/sthn.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-forum/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-forum/tgn.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-github/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-github/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-github/recurrencybaseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-github/recurrencybaseline.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-github/run_seeds.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-github/run_seeds.sh -------------------------------------------------------------------------------- /examples/linkproppred/thgl-github/sthn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-github/sthn.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-github/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-github/tgn.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-myket/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-myket/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-myket/recurrencybaseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-myket/recurrencybaseline.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-myket/sthn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-myket/sthn.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-myket/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-myket/tgn.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-software/STHN_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-software/STHN_README.md -------------------------------------------------------------------------------- /examples/linkproppred/thgl-software/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-software/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-software/recurrencybaseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-software/recurrencybaseline.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-software/sthn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-software/sthn.py -------------------------------------------------------------------------------- /examples/linkproppred/thgl-software/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/thgl-software/tgn.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-icews/cen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-icews/cen.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-icews/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-icews/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-icews/recurrencybaseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-icews/recurrencybaseline.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-icews/regcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-icews/regcn.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-icews/timetraveler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-icews/timetraveler.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-icews/tkgl-icews_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-icews/tkgl-icews_example.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-icews/tlogic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-icews/tlogic.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-polecat/cen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-polecat/cen.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-polecat/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-polecat/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-polecat/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-polecat/example.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-polecat/recurrencybaseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-polecat/recurrencybaseline.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-polecat/regcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-polecat/regcn.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-polecat/timetraveler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-polecat/timetraveler.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-polecat/tkgl-polecat_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-polecat/tkgl-polecat_example.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-polecat/tlogic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-polecat/tlogic.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-smallpedia/cen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-smallpedia/cen.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-smallpedia/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-smallpedia/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-smallpedia/recurrencybaseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-smallpedia/recurrencybaseline.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-smallpedia/regcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-smallpedia/regcn.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-smallpedia/timetraveler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-smallpedia/timetraveler.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-smallpedia/tkgl-smallpedia_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-smallpedia/tkgl-smallpedia_example.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-smallpedia/tlogic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-smallpedia/tlogic.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-wikidata/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-wikidata/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-wikidata/recurrencybaseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-wikidata/recurrencybaseline.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-wikidata/regcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-wikidata/regcn.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-wikidata/tkgl-wikidata_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-wikidata/tkgl-wikidata_example.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-wikidata/tlogic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-wikidata/tlogic.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-yago/cen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-yago/cen.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-yago/edgebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-yago/edgebank.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-yago/recurrencybaseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-yago/recurrencybaseline.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-yago/regcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-yago/regcn.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-yago/timetraveler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-yago/timetraveler.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-yago/tkgl-yago_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-yago/tkgl-yago_example.py -------------------------------------------------------------------------------- /examples/linkproppred/tkgl-yago/tlogic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/linkproppred/tkgl-yago/tlogic.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-genre/dyrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-genre/dyrep.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-genre/moving_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-genre/moving_average.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-genre/persistant_forecast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-genre/persistant_forecast.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-genre/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-genre/tgn.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-reddit/dyrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-reddit/dyrep.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-reddit/moving_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-reddit/moving_average.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-reddit/persistant_forecast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-reddit/persistant_forecast.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-reddit/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-reddit/tgn.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-token/dyrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-token/dyrep.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-token/moving_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-token/moving_average.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-token/persistant_forecast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-token/persistant_forecast.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-token/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-token/tgn.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-trade/count_new_nodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-trade/count_new_nodes.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-trade/dyrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-trade/dyrep.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-trade/moving_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-trade/moving_average.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-trade/persistant_forecast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-trade/persistant_forecast.py -------------------------------------------------------------------------------- /examples/nodeproppred/tgbn-trade/tgn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/examples/nodeproppred/tgbn-trade/tgn.py -------------------------------------------------------------------------------- /imgs/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/imgs/logo.png -------------------------------------------------------------------------------- /imgs/pipeline.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/imgs/pipeline.PNG -------------------------------------------------------------------------------- /imgs/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/imgs/pipeline.png -------------------------------------------------------------------------------- /imgs/stats.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/imgs/stats.png -------------------------------------------------------------------------------- /mkdocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/mkdocs.yml -------------------------------------------------------------------------------- /modules/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/decoder.py -------------------------------------------------------------------------------- /modules/early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/early_stopping.py -------------------------------------------------------------------------------- /modules/edgebank_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/edgebank_predictor.py -------------------------------------------------------------------------------- /modules/emb_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/emb_module.py -------------------------------------------------------------------------------- /modules/heuristics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/heuristics.py -------------------------------------------------------------------------------- /modules/memory_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/memory_module.py -------------------------------------------------------------------------------- /modules/msg_agg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/msg_agg.py -------------------------------------------------------------------------------- /modules/msg_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/msg_func.py -------------------------------------------------------------------------------- /modules/neighbor_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/neighbor_loader.py -------------------------------------------------------------------------------- /modules/nodebank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/nodebank.py -------------------------------------------------------------------------------- /modules/recurrencybaseline_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/recurrencybaseline_predictor.py -------------------------------------------------------------------------------- /modules/rgcn_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/rgcn_layers.py -------------------------------------------------------------------------------- /modules/rgcn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/rgcn_model.py -------------------------------------------------------------------------------- /modules/rrgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/rrgcn.py -------------------------------------------------------------------------------- /modules/sampler_core.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/sampler_core.cpp -------------------------------------------------------------------------------- /modules/sthn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/sthn.py -------------------------------------------------------------------------------- /modules/sthn_sampler_setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/sthn_sampler_setup.py -------------------------------------------------------------------------------- /modules/time_enc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/time_enc.py -------------------------------------------------------------------------------- /modules/timetraveler_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/timetraveler_agent.py -------------------------------------------------------------------------------- /modules/timetraveler_dirichlet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/timetraveler_dirichlet.py -------------------------------------------------------------------------------- /modules/timetraveler_environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/timetraveler_environment.py -------------------------------------------------------------------------------- /modules/timetraveler_episode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/timetraveler_episode.py -------------------------------------------------------------------------------- /modules/timetraveler_policygradient.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/timetraveler_policygradient.py -------------------------------------------------------------------------------- /modules/timetraveler_trainertester.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/timetraveler_trainertester.py -------------------------------------------------------------------------------- /modules/tkg_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/tkg_utils.py -------------------------------------------------------------------------------- /modules/tkg_utils_dgl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/tkg_utils_dgl.py -------------------------------------------------------------------------------- /modules/tlogic_apply_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/tlogic_apply_modules.py -------------------------------------------------------------------------------- /modules/tlogic_learn_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/modules/tlogic_learn_modules.py -------------------------------------------------------------------------------- /pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/pipeline.png -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements-dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/requirements-dev.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/run.sh -------------------------------------------------------------------------------- /scripts/env.sh: -------------------------------------------------------------------------------- 1 | module load python/3.9 2 | source $HOME/tgbenv/bin/activate 3 | 4 | -------------------------------------------------------------------------------- /scripts/mila.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/scripts/mila.sh -------------------------------------------------------------------------------- /scripts/mila_install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/scripts/mila_install.sh -------------------------------------------------------------------------------- /scripts/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/scripts/run.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/setup.py -------------------------------------------------------------------------------- /stats_figures/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/stats_figures/README.md -------------------------------------------------------------------------------- /stats_figures/compute_dataset_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/stats_figures/compute_dataset_stats.py -------------------------------------------------------------------------------- /stats_figures/compute_relation_dataset_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/stats_figures/compute_relation_dataset_stats.py -------------------------------------------------------------------------------- /stats_figures/compute_relation_results_df.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/stats_figures/compute_relation_results_df.py -------------------------------------------------------------------------------- /stats_figures/create_edges_figures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/stats_figures/create_edges_figures.py -------------------------------------------------------------------------------- /stats_figures/create_relation_figures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/stats_figures/create_relation_figures.py -------------------------------------------------------------------------------- /stats_figures/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/stats_figures/dataset_utils.py -------------------------------------------------------------------------------- /tgb/datasets/ICEWS14/ent2word.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/ICEWS14/ent2word.py -------------------------------------------------------------------------------- /tgb/datasets/ICEWS14/icews14.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/ICEWS14/icews14.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/MAG/mag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/MAG/mag.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/MAG/old/plot_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/MAG/old/plot_stats.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/dgraph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/dgraph.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/dgraph_Readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/dgraph_Readme.md -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/process_arxiv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/process_arxiv.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/process_github.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/process_github.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/tgbl-coin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/tgbl-coin.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/tgbl-coin_neg_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/tgbl-coin_neg_generator.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/tgbl-comment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/tgbl-comment.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/tgbl-comment_neg_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/tgbl-comment_neg_generator.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/tgbl-flight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/tgbl-flight.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/tgbl-flight_neg_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/tgbl-flight_neg_generator.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/tgbl-review.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/tgbl-review.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/tgbl-review_neg_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/tgbl-review_neg_generator.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/tgbl-wiki_neg_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/tgbl-wiki_neg_generator.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/tgbn-genre.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/tgbn-genre.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/tgbn-reddit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/tgbn-reddit.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/tgbn-token.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/tgbn-token.py -------------------------------------------------------------------------------- /tgb/datasets/dataset_scripts/tgbn-trade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/dataset_scripts/tgbn-trade.py -------------------------------------------------------------------------------- /tgb/datasets/tgbl_enron/tgbl-enron_neg_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tgbl_enron/tgbl-enron_neg_generator.py -------------------------------------------------------------------------------- /tgb/datasets/tgbl_enron/tgbl_enron.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tgbl_enron/tgbl_enron.py -------------------------------------------------------------------------------- /tgb/datasets/tgbl_lastfm/tgbl-lastfm_neg_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tgbl_lastfm/tgbl-lastfm_neg_generator.py -------------------------------------------------------------------------------- /tgb/datasets/tgbl_subreddit/tgbl-subreddit_neg_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tgbl_subreddit/tgbl-subreddit_neg_generator.py -------------------------------------------------------------------------------- /tgb/datasets/tgbl_uci/tgbl-uci_neg_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tgbl_uci/tgbl-uci_neg_generator.py -------------------------------------------------------------------------------- /tgb/datasets/tgbl_uci/tgbl_uci.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tgbl_uci/tgbl_uci.py -------------------------------------------------------------------------------- /tgb/datasets/thgl_forum/merge_files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/thgl_forum/merge_files.py -------------------------------------------------------------------------------- /tgb/datasets/thgl_forum/num_edges_discretized_thgl-forum.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/thgl_forum/num_edges_discretized_thgl-forum.png -------------------------------------------------------------------------------- /tgb/datasets/thgl_forum/thgl-forum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/thgl_forum/thgl-forum.py -------------------------------------------------------------------------------- /tgb/datasets/thgl_forum/thgl_forum_ns_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/thgl_forum/thgl_forum_ns_gen.py -------------------------------------------------------------------------------- /tgb/datasets/thgl_github/2024_01/github_extract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/thgl_github/2024_01/github_extract.py -------------------------------------------------------------------------------- /tgb/datasets/thgl_github/2024_02/github_extract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/thgl_github/2024_02/github_extract.py -------------------------------------------------------------------------------- /tgb/datasets/thgl_github/2024_03/github_extract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/thgl_github/2024_03/github_extract.py -------------------------------------------------------------------------------- /tgb/datasets/thgl_github/extract_subset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/thgl_github/extract_subset.py -------------------------------------------------------------------------------- /tgb/datasets/thgl_github/thgl_github.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/thgl_github/thgl_github.py -------------------------------------------------------------------------------- /tgb/datasets/thgl_github/thgl_github_ns_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/thgl_github/thgl_github_ns_gen.py -------------------------------------------------------------------------------- /tgb/datasets/thgl_myket/thgl_myket.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/thgl_myket/thgl_myket.py -------------------------------------------------------------------------------- /tgb/datasets/thgl_myket/thgl_myket_ns_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/thgl_myket/thgl_myket_ns_gen.py -------------------------------------------------------------------------------- /tgb/datasets/thgl_software/thgl_software.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/thgl_software/thgl_software.py -------------------------------------------------------------------------------- /tgb/datasets/thgl_software/thgl_software_ns_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/thgl_software/thgl_software_ns_gen.py -------------------------------------------------------------------------------- /tgb/datasets/tkgl_icews/tkgl_icews.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tkgl_icews/tkgl_icews.py -------------------------------------------------------------------------------- /tgb/datasets/tkgl_icews/tkgl_icews_ns_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tkgl_icews/tkgl_icews_ns_gen.py -------------------------------------------------------------------------------- /tgb/datasets/tkgl_polecat/tkgl_polecat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tkgl_polecat/tkgl_polecat.py -------------------------------------------------------------------------------- /tgb/datasets/tkgl_polecat/tkgl_polecat_ns_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tkgl_polecat/tkgl_polecat_ns_gen.py -------------------------------------------------------------------------------- /tgb/datasets/tkgl_smallpedia/smallpedia_remove_conflict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tkgl_smallpedia/smallpedia_remove_conflict.py -------------------------------------------------------------------------------- /tgb/datasets/tkgl_smallpedia/tkgl_smallpedia_ns_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tkgl_smallpedia/tkgl_smallpedia_ns_gen.py -------------------------------------------------------------------------------- /tgb/datasets/tkgl_wikidata/extract.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tkgl_wikidata/extract.sh -------------------------------------------------------------------------------- /tgb/datasets/tkgl_wikidata/time_edges/tkgl-wikidata_extract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tkgl_wikidata/time_edges/tkgl-wikidata_extract.py -------------------------------------------------------------------------------- /tgb/datasets/tkgl_wikidata/tkgl-wikidata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tkgl_wikidata/tkgl-wikidata.py -------------------------------------------------------------------------------- /tgb/datasets/tkgl_wikidata/tkgl_wikidata_mining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tkgl_wikidata/tkgl_wikidata_mining.py -------------------------------------------------------------------------------- /tgb/datasets/tkgl_wikidata/tkgl_wikidata_ns_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tkgl_wikidata/tkgl_wikidata_ns_gen.py -------------------------------------------------------------------------------- /tgb/datasets/tkgl_wikidata/wikidata_remove_conflict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tkgl_wikidata/wikidata_remove_conflict.py -------------------------------------------------------------------------------- /tgb/datasets/tkgl_yago/tkgl_yago.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tkgl_yago/tkgl_yago.py -------------------------------------------------------------------------------- /tgb/datasets/tkgl_yago/tkgl_yago_ns_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/datasets/tkgl_yago/tkgl_yago_ns_gen.py -------------------------------------------------------------------------------- /tgb/linkproppred/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/linkproppred/dataset.py -------------------------------------------------------------------------------- /tgb/linkproppred/dataset_pyg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/linkproppred/dataset_pyg.py -------------------------------------------------------------------------------- /tgb/linkproppred/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/linkproppred/evaluate.py -------------------------------------------------------------------------------- /tgb/linkproppred/negative_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/linkproppred/negative_generator.py -------------------------------------------------------------------------------- /tgb/linkproppred/negative_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/linkproppred/negative_sampler.py -------------------------------------------------------------------------------- /tgb/linkproppred/thg_negative_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/linkproppred/thg_negative_generator.py -------------------------------------------------------------------------------- /tgb/linkproppred/thg_negative_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/linkproppred/thg_negative_sampler.py -------------------------------------------------------------------------------- /tgb/linkproppred/tkg_negative_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/linkproppred/tkg_negative_generator.py -------------------------------------------------------------------------------- /tgb/linkproppred/tkg_negative_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/linkproppred/tkg_negative_sampler.py -------------------------------------------------------------------------------- /tgb/nodeproppred/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/nodeproppred/dataset.py -------------------------------------------------------------------------------- /tgb/nodeproppred/dataset_pyg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/nodeproppred/dataset_pyg.py -------------------------------------------------------------------------------- /tgb/nodeproppred/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/nodeproppred/evaluate.py -------------------------------------------------------------------------------- /tgb/utils/dataset_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/utils/dataset_stats.py -------------------------------------------------------------------------------- /tgb/utils/info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/utils/info.py -------------------------------------------------------------------------------- /tgb/utils/pre_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/utils/pre_process.py -------------------------------------------------------------------------------- /tgb/utils/stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/utils/stats.py -------------------------------------------------------------------------------- /tgb/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shenyangHuang/TGB/HEAD/tgb/utils/utils.py --------------------------------------------------------------------------------