This is the implementation of the paper Explain and Predict, and then Predict Again (accepted in WSDM2021). This code is implemented based on the pipeline model of the Eraserbenchmark. All data used by the model can be found from the Eraser Benchmark, too.
- Install the required packages from the
requirements.txt
bypip install -r requirements.txt
- The implementation entry is under
expred/train
. To run the training, simply copy and paste the following commands:export PYTHONPATH=$PYTHONPATH:./ && python expred/train.py --data_dir /dir/to/your/datasets/{movies,fever,multirc} --output_dir /dir/to/your/trained_data --conf ./params/{movies,fever,multirc}_expred.json
Not that depending on your hardware you may have to change the batch_size
in the config file.