/representation-causal-public

Primary LanguageJupyter NotebookMIT LicenseMIT

Reference implementation for "Desiderata for Representation Learning: A Causal Perspective"

This repository contains the code for

  • the empirical study about CAUSAL-REP (Section 2.4)

  • the empirical study about IOSS (Section 3.4)

Environment

The code is tested in Python 3.8.6 and Pytorch 1.4.0.

References

Wang, Y., & Jordan, M.I. (2021) Desiderata for Representation Learning: A Causal Perspective. arXiv preprint arXiv:2109.03795. [link]