/hybrid-causal-discovery

Source code of AAAI'22 paper: A Hybrid Causal Structure Learning Algorithm for Mixed-type Data

Primary LanguagePythonMIT LicenseMIT

Souce code of HCM (AAAI 2022 paper)

  • Yan Li, Rui Xia, Chunchen Liu, and Liang Sun. 2022. A Hybrid Causal Structure Learning Algorithm for Mixed-type Data. In Proc. of the AAAI Conference on Artificial Intelligence (AAAI 2022), 2022. [paper]

Requirements

  1. Install required R package

    • Require R >= 3.6.3
    • Run following code in the R Console

    install.packages("/ossfs/workspace/pai-algo-dev/momentchi2_0.1.5.tar", repos=NULL, type="source")

  2. Install required python packages

    pip install requirements.txt

Run example

  • run.py provides an example for you to conduct one experiment and one can refer to const.py to learn the meaning of parameters.

Generate simulation data

  • please go mvpsimulation/ folder, and the README.md file inside the folder for more information.

Citation

If you find this repo useful, please cite our paper.

@inproceedings{li2022hcm,
  title={A Hybrid Causal Structure Learning Algorithm for Mixed-type Data},
  author={Li, Yan and Xia, Rui and Liu, Chunchen and Sun, Liang},
  booktitle={Proceedings of the AAAI Conference on Artificial Intelligence (AAAI 2022)},
  location={},
  pages={},
  year={2022} 
}