ECRT: Energy-based Causal Representation Transfer
Code for Supercharging Imbalanced Data Learning With Energy-basedContrastive Representation Transfer (NeurIPS 2021)
Model
Prerequisites
The algorithm is built with:
- Python (version 3.7 or higher)
- Numpy (version 1.16 or higher)
- PyTorch (version 1.3.1)
Clone the repository, e.g.:
git clone https://github.com/ZidiXiu/ECRT.git
Running the Toy Dataset
Here we present a toy synthetic dataset
python train train_CRT_toy.py
When building the CRT framework, we referenced the following sources: