├── .gitignore ├── DG_data ├── DATASETS_README.md └── myket │ └── myket.csv ├── LICENSE ├── README.md ├── evaluate_link_prediction.py ├── evaluate_models_utils.py ├── evaluate_node_classification.py ├── figures ├── DyGFormer_framework.jpg └── DyGLib_procedure.jpg ├── models ├── CAWN.py ├── DyGFormer.py ├── EdgeBank.py ├── GraphMixer.py ├── MemoryModel.py ├── TCL.py ├── TGAT.py └── modules.py ├── preprocess_data ├── data_statistics.py ├── preprocess_all_data.py └── preprocess_data.py ├── processed_data ├── .gitkeep └── myket │ ├── ml_myket.csv │ ├── ml_myket.npy │ └── ml_myket_node.npy ├── requirements.txt ├── train_link_prediction.py ├── train_node_classification.py └── utils ├── DataLoader.py ├── EarlyStopping.py ├── load_configs.py ├── metrics.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/.gitignore -------------------------------------------------------------------------------- /DG_data/DATASETS_README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/DG_data/DATASETS_README.md -------------------------------------------------------------------------------- /DG_data/myket/myket.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/DG_data/myket/myket.csv -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/README.md -------------------------------------------------------------------------------- /evaluate_link_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/evaluate_link_prediction.py -------------------------------------------------------------------------------- /evaluate_models_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/evaluate_models_utils.py -------------------------------------------------------------------------------- /evaluate_node_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/evaluate_node_classification.py -------------------------------------------------------------------------------- /figures/DyGFormer_framework.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/figures/DyGFormer_framework.jpg -------------------------------------------------------------------------------- /figures/DyGLib_procedure.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/figures/DyGLib_procedure.jpg -------------------------------------------------------------------------------- /models/CAWN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/models/CAWN.py -------------------------------------------------------------------------------- /models/DyGFormer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/models/DyGFormer.py -------------------------------------------------------------------------------- /models/EdgeBank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/models/EdgeBank.py -------------------------------------------------------------------------------- /models/GraphMixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/models/GraphMixer.py -------------------------------------------------------------------------------- /models/MemoryModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/models/MemoryModel.py -------------------------------------------------------------------------------- /models/TCL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/models/TCL.py -------------------------------------------------------------------------------- /models/TGAT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/models/TGAT.py -------------------------------------------------------------------------------- /models/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/models/modules.py -------------------------------------------------------------------------------- /preprocess_data/data_statistics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/preprocess_data/data_statistics.py -------------------------------------------------------------------------------- /preprocess_data/preprocess_all_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/preprocess_data/preprocess_all_data.py -------------------------------------------------------------------------------- /preprocess_data/preprocess_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/preprocess_data/preprocess_data.py -------------------------------------------------------------------------------- /processed_data/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /processed_data/myket/ml_myket.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/processed_data/myket/ml_myket.csv -------------------------------------------------------------------------------- /processed_data/myket/ml_myket.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/processed_data/myket/ml_myket.npy -------------------------------------------------------------------------------- /processed_data/myket/ml_myket_node.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/processed_data/myket/ml_myket_node.npy -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch>=1.8.1 2 | numpy 3 | pandas 4 | tqdm 5 | tabulate -------------------------------------------------------------------------------- /train_link_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/train_link_prediction.py -------------------------------------------------------------------------------- /train_node_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/train_node_classification.py -------------------------------------------------------------------------------- /utils/DataLoader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/utils/DataLoader.py -------------------------------------------------------------------------------- /utils/EarlyStopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/utils/EarlyStopping.py -------------------------------------------------------------------------------- /utils/load_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/utils/load_configs.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yule-BUAA/DyGLib/HEAD/utils/utils.py --------------------------------------------------------------------------------