Skip to content

ignavierng/notears-tensorflow

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

NOTEARS-Tensorflow

This repository is a Tensorflow 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

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.

About

Reimplementation of NOTEARS in Tensorflow

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published