English

gCastle: A Python Toolbox for Causal Discovery

Machine Learning 2021-12-01 v1 Machine Learning

Abstract

gCastle\texttt{gCastle} is an end-to-end Python toolbox for causal structure learning. It provides functionalities of generating data from either simulator or real-world dataset, learning causal structure from the data, and evaluating the learned graph, together with useful practices such as prior knowledge insertion, preliminary neighborhood selection, and post-processing to remove false discoveries. Compared with related packages, gCastle\texttt{gCastle} includes many recently developed gradient-based causal discovery methods with optional GPU acceleration. gCastle\texttt{gCastle} brings convenience to researchers who may directly experiment with the code as well as practitioners with graphical user interference. Three real-world datasets in telecommunications are also provided in the current version. gCastle\texttt{gCastle} is available under Apache License 2.0 at \url{https://github.com/huawei-noah/trustworthyAI/tree/master/gcastle}.

Keywords

Cite

@article{arxiv.2111.15155,
  title  = {gCastle: A Python Toolbox for Causal Discovery},
  author = {Keli Zhang and Shengyu Zhu and Marcus Kalander and Ignavier Ng and Junjian Ye and Zhitang Chen and Lujia Pan},
  journal= {arXiv preprint arXiv:2111.15155},
  year   = {2021}
}

Comments

Tech report describing the gCastle toolbox. More details can be found in the github repository https://github.com/huawei-noah/trustworthyAI/tree/master/gcastle

R2 v1 2026-06-24T07:57:10.152Z