├── .gitignore ├── LICENSE ├── README.md ├── figures └── fp_diagram.png ├── hyperparameters.yaml ├── src ├── __init__.py ├── baselines │ ├── gcn_mf.py │ └── pa_gnn.py ├── data_loading.py ├── data_utils.py ├── evaluation.py ├── feature_propagation.py ├── filling_strategies.py ├── models.py ├── run.py ├── seeds.py ├── train.py └── utils.py └── tests ├── feature_propagation_test.py ├── filling_strategies_test.py └── utils_test.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/README.md -------------------------------------------------------------------------------- /figures/fp_diagram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/figures/fp_diagram.png -------------------------------------------------------------------------------- /hyperparameters.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/hyperparameters.yaml -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/baselines/gcn_mf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/src/baselines/gcn_mf.py -------------------------------------------------------------------------------- /src/baselines/pa_gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/src/baselines/pa_gnn.py -------------------------------------------------------------------------------- /src/data_loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/src/data_loading.py -------------------------------------------------------------------------------- /src/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/src/data_utils.py -------------------------------------------------------------------------------- /src/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/src/evaluation.py -------------------------------------------------------------------------------- /src/feature_propagation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/src/feature_propagation.py -------------------------------------------------------------------------------- /src/filling_strategies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/src/filling_strategies.py -------------------------------------------------------------------------------- /src/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/src/models.py -------------------------------------------------------------------------------- /src/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/src/run.py -------------------------------------------------------------------------------- /src/seeds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/src/seeds.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/src/train.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/src/utils.py -------------------------------------------------------------------------------- /tests/feature_propagation_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/tests/feature_propagation_test.py -------------------------------------------------------------------------------- /tests/filling_strategies_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/tests/filling_strategies_test.py -------------------------------------------------------------------------------- /tests/utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/twitter-research/feature-propagation/HEAD/tests/utils_test.py --------------------------------------------------------------------------------