├── README.md ├── data └── __init__.py ├── environment.yml ├── figures ├── OTGNet.png └── teaser.png ├── models ├── IB.py ├── OTGNet.py ├── TimeEncoder.py ├── __init__.py └── pseudo_gen.py ├── modules ├── __init__.py ├── buffer.py └── memory.py ├── process_raw_data ├── reddit │ ├── preprocess.py │ └── process_csv.py ├── taobao │ ├── pre_process.py │ └── process_csv.py └── yelp │ ├── generate_graph.py │ ├── pre_process_yelp.py │ └── process_csv.py ├── run.py ├── train.py └── utils ├── __init__.py ├── data_processing.py ├── evaluation.py ├── log_and_checkpoints.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/README.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/environment.yml -------------------------------------------------------------------------------- /figures/OTGNet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/figures/OTGNet.png -------------------------------------------------------------------------------- /figures/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/figures/teaser.png -------------------------------------------------------------------------------- /models/IB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/models/IB.py -------------------------------------------------------------------------------- /models/OTGNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/models/OTGNet.py -------------------------------------------------------------------------------- /models/TimeEncoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/models/TimeEncoder.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/pseudo_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/models/pseudo_gen.py -------------------------------------------------------------------------------- /modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modules/buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/modules/buffer.py -------------------------------------------------------------------------------- /modules/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/modules/memory.py -------------------------------------------------------------------------------- /process_raw_data/reddit/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/process_raw_data/reddit/preprocess.py -------------------------------------------------------------------------------- /process_raw_data/reddit/process_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/process_raw_data/reddit/process_csv.py -------------------------------------------------------------------------------- /process_raw_data/taobao/pre_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/process_raw_data/taobao/pre_process.py -------------------------------------------------------------------------------- /process_raw_data/taobao/process_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/process_raw_data/taobao/process_csv.py -------------------------------------------------------------------------------- /process_raw_data/yelp/generate_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/process_raw_data/yelp/generate_graph.py -------------------------------------------------------------------------------- /process_raw_data/yelp/pre_process_yelp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/process_raw_data/yelp/pre_process_yelp.py -------------------------------------------------------------------------------- /process_raw_data/yelp/process_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/process_raw_data/yelp/process_csv.py -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/run.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/data_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/utils/data_processing.py -------------------------------------------------------------------------------- /utils/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/utils/evaluation.py -------------------------------------------------------------------------------- /utils/log_and_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/utils/log_and_checkpoints.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tulerfeng/OTGNet/HEAD/utils/utils.py --------------------------------------------------------------------------------