NOTEARS-Tensorflow2

This repository is a Tensorflow2 reimplementation of NOTEARS [1].

1. Setup

pip install -r requirements.txt

2. Training

To run NoTears, for example, run:

python main.py  --seed 1230 \
                --d 20 \
                --n 1000 \
                --degree 4 \
                --graph_thres 0.3 \
                --l1_lambda 0.1

Remark

• Some of the code implementation is referred from https://github.com/xunzheng/notears

References

[1] Zheng, X., Aragam, B., Ravikumar, P., and Xing, E. P. DAGs with NO TEARS: Continuous optimization for structure learning. In Advances in Neural Information Processing Systems, 2018.