├── .gitignore ├── LICENSE ├── README.md ├── conda_env.yml ├── config ├── imputation │ ├── brits.yaml │ ├── grin.yaml │ ├── saits.yaml │ ├── spin.yaml │ ├── spin_h.yaml │ └── transformer.yaml └── inference.yaml ├── experiments ├── run_imputation.py └── run_inference.py ├── paper_neurips.pdf ├── poster_neurips.pdf ├── sparse_att.png ├── spin ├── __init__.py ├── baselines │ ├── __init__.py │ ├── brits │ │ ├── __init__.py │ │ ├── brits.py │ │ └── layers.py │ ├── saits │ │ ├── __init__.py │ │ ├── layers.py │ │ └── saits.py │ └── transformer │ │ ├── __init__.py │ │ └── transformer.py ├── imputers │ ├── __init__.py │ ├── brits_imputer.py │ ├── saits_imputer.py │ └── spin_imputer.py ├── layers │ ├── __init__.py │ ├── additive_attention.py │ ├── hierarchical_temporal_graph_attention.py │ ├── postional_encoding.py │ └── temporal_graph_additive_attention.py ├── models │ ├── __init__.py │ ├── spin.py │ └── spin_hierarchical.py ├── scheduler.py └── utils.py └── tsl_config.yaml /.gitignore: -------------------------------------------------------------------------------- 1 | *.DS_STORE 2 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/README.md -------------------------------------------------------------------------------- /conda_env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/conda_env.yml -------------------------------------------------------------------------------- /config/imputation/brits.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/config/imputation/brits.yaml -------------------------------------------------------------------------------- /config/imputation/grin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/config/imputation/grin.yaml -------------------------------------------------------------------------------- /config/imputation/saits.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/config/imputation/saits.yaml -------------------------------------------------------------------------------- /config/imputation/spin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/config/imputation/spin.yaml -------------------------------------------------------------------------------- /config/imputation/spin_h.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/config/imputation/spin_h.yaml -------------------------------------------------------------------------------- /config/imputation/transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/config/imputation/transformer.yaml -------------------------------------------------------------------------------- /config/inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/config/inference.yaml -------------------------------------------------------------------------------- /experiments/run_imputation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/experiments/run_imputation.py -------------------------------------------------------------------------------- /experiments/run_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/experiments/run_inference.py -------------------------------------------------------------------------------- /paper_neurips.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/paper_neurips.pdf -------------------------------------------------------------------------------- /poster_neurips.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/poster_neurips.pdf -------------------------------------------------------------------------------- /sparse_att.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/sparse_att.png -------------------------------------------------------------------------------- /spin/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /spin/baselines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/baselines/__init__.py -------------------------------------------------------------------------------- /spin/baselines/brits/__init__.py: -------------------------------------------------------------------------------- 1 | from .brits import BRITS 2 | -------------------------------------------------------------------------------- /spin/baselines/brits/brits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/baselines/brits/brits.py -------------------------------------------------------------------------------- /spin/baselines/brits/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/baselines/brits/layers.py -------------------------------------------------------------------------------- /spin/baselines/saits/__init__.py: -------------------------------------------------------------------------------- 1 | from .saits import SAITS 2 | -------------------------------------------------------------------------------- /spin/baselines/saits/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/baselines/saits/layers.py -------------------------------------------------------------------------------- /spin/baselines/saits/saits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/baselines/saits/saits.py -------------------------------------------------------------------------------- /spin/baselines/transformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/baselines/transformer/__init__.py -------------------------------------------------------------------------------- /spin/baselines/transformer/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/baselines/transformer/transformer.py -------------------------------------------------------------------------------- /spin/imputers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/imputers/__init__.py -------------------------------------------------------------------------------- /spin/imputers/brits_imputer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/imputers/brits_imputer.py -------------------------------------------------------------------------------- /spin/imputers/saits_imputer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/imputers/saits_imputer.py -------------------------------------------------------------------------------- /spin/imputers/spin_imputer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/imputers/spin_imputer.py -------------------------------------------------------------------------------- /spin/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/layers/__init__.py -------------------------------------------------------------------------------- /spin/layers/additive_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/layers/additive_attention.py -------------------------------------------------------------------------------- /spin/layers/hierarchical_temporal_graph_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/layers/hierarchical_temporal_graph_attention.py -------------------------------------------------------------------------------- /spin/layers/postional_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/layers/postional_encoding.py -------------------------------------------------------------------------------- /spin/layers/temporal_graph_additive_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/layers/temporal_graph_additive_attention.py -------------------------------------------------------------------------------- /spin/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/models/__init__.py -------------------------------------------------------------------------------- /spin/models/spin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/models/spin.py -------------------------------------------------------------------------------- /spin/models/spin_hierarchical.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/models/spin_hierarchical.py -------------------------------------------------------------------------------- /spin/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/scheduler.py -------------------------------------------------------------------------------- /spin/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/spin/utils.py -------------------------------------------------------------------------------- /tsl_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Graph-Machine-Learning-Group/spin/HEAD/tsl_config.yaml --------------------------------------------------------------------------------