# On the Discrimination Risk of Mean Aggregation Feature Imputation in Graphs 

This repository is the official implementation of On the Discrimination Risk of Mean Aggregation Feature Imputation in Graphs. 

## Requirements

To install requirements, use the provided Dockerfile:

```setup
docker build -t "pyg:discrimination_risk" - < Dockerfile
```

## Training and Evaluation

To run all the experiment(s) in the paper (i.e. imputation, training, evaluation, displaying results), simply run all the cells in `run.py`. You can change hyperparameters by manually modifying the `ArgumentParser` arguments in `run.py`.

## License

We adapted the code for data utilities, Feature Propagation, and model training from the [Feature Propagation paper](https://github.com/twitter-research/feature-propagation). All other code is written by the authors.


