This repository contains training code for baselines and methods implemented in the paper
python -m pip install -r requirements.txt
Tested with transE, distMult, CompIEx, and HolE embeddings
python main.py --datapath data --datafile CLEVRER_data.pkl --baseline trans_e
python main.py --datapath data --datafile CLEVRER_data.pkl --baseline dm_e
python main.py --datapath data --datafile CLEVRER_data.pkl --baseline comp_e
python main.py --datapath data --datafile CLEVRER_data.pkl --baseline hyp_e
python main.py --datapath data --datafile CLEVRER_H_data.pkl --baseline trans_e
python main.py --datapath data --datafile CLEVRER_H_data.pkl --baseline dm_e
python main.py --datapath data --datafile CLEVRER_H_data.pkl --baseline comp_e
python main.py --datapath data --datafile CLEVRER_H_data.pkl --baseline hyp_e
Tested with both the CLEVRER and CLEVRER Humans Dataset
python main.py --datapath data --datafile CLEVRER_data.pkl
python main.py --datapath data --datafile CLEVRER_H_data.pkl