Gradient-Based Neural DAG Learning

This code was written by the authors of the ICLR 2020 submission: "Gradient-Based Neural DAG Learning". Our implementation is in PyTorch but some functions rely on the Causal Discovery Toolbox which relies partly on the R programming language.

Run the code

To use our implementation of GraN-DAG, simply install Singularity (instructions: https://www.sylabs.io/guides/3.0/user-guide/installation.html) and run the code in our container (download it here: https://zenodo.org/record/7700409). Use start_example.sh (update the paths) to launch the differents methods (GraN-DAG, DAG-GNN, NOTEARS, CAM).

About

MIT License

Languages

Language:C++ 80.7%Language:Fortran 9.9%Language:C 3.2%Language:CMake 2.5%Language:Python 1.5%Language:Cuda 1.1%Language:Jupyter Notebook 0.4%Language:Shell 0.2%Language:R 0.2%Language:HTML 0.1%Language:JavaScript 0.1%Language:CSS 0.0%