implementation of paper Fair Representation Learning: An Alternative to Mutual Information (KDD 2022)
- python >= 3.6
- pytorch >= 1.4
For dataset Adult-sex with default parameters
python FairDisCo.py --cuda 0 --data Adult-sex
For other dataset, you can change the data parameter
python FairDisCo.py --cuda 0 --data Adult-sex/Adult-race/Health/Bank/German/Creadit/Student/Compas-sex/Compas-race
For dataset MNIST or MNIST-color with default parameters
python FairDisCoImage.py --cuda 0 --data color/number
file | description |
---|---|
data/ | The datasets dir |
res/ | The results dir |
FairDisCo.py | The training code of fairdisco |
FairDisCoImage.py | The training code of fairdisco with image data |
module.py | The definition of fairdisco's model structure |
utils.py | The data loading function and evaluation function |