Official code for the paper Dual Algorithmic Reasoning
reqs/*
contains requirements files for linux-cpu
, macos-cpu
and linux-gpu
The two main files build_data.py
and run.py
have a Typer CLI. Refer to each script's commands by running
python <filename>.py --help
python build_data.py [PARAMS]
creates training/validation/test data and should be the first command you run.
python run.py [COMMAND] [PARAMS]
runs either the model selection (valid
) or the risk assessment (test
).
config/*
contains configurations for data generation, model hyperparameters (included the optimal ones) and general configs.config/exp/*
contains the YAML files for reproducing the experiments shown in the paper (use the*.one.yaml
versions).nn/*
contains all the model implementations.