Metadata-Version: 2.4
Name: causal_toolkit_qingwei
Version: 0.1.5
Summary: A Python package for causal inference methods including ATE estimation, propensity score methods, and meta-learners
Author-email: Qingwei Zhang <qzhang.canada@gmail.com>
License: MIT
Project-URL: Homepage, https://github.com/qingweizhang054/causal-toolkit-qingwei-package
Project-URL: Documentation, https://github.com/qingweizhang054/causal-toolkit-qingwei-package#readme
Project-URL: Repository, https://github.com/qingweizhang054/causal-toolkit-qingwei-package
Project-URL: Bug Tracker, https://github.com/qingweizhang054/causal-toolkit-qingwei-package/issues
Keywords: causal inference,statistics,machine learning,treatment effects
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Topic :: Scientific/Engineering :: Mathematics
Classifier: Topic :: Scientific/Engineering :: Information Analysis
Requires-Python: >=3.8
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: packaging>=24.0
Requires-Dist: pandas>=1.3.0
Requires-Dist: numpy>=1.21.0
Requires-Dist: scipy>=1.7.0
Requires-Dist: scikit-learn>=1.0.0
Requires-Dist: lightgbm>=3.3.0
Requires-Dist: patsy>=0.5.0
Provides-Extra: dev
Requires-Dist: pytest>=7.0.0; extra == "dev"
Requires-Dist: pytest-cov>=3.0.0; extra == "dev"
Requires-Dist: black>=22.0.0; extra == "dev"
Requires-Dist: pylint>=2.12.0; extra == "dev"
Requires-Dist: mypy>=0.950; extra == "dev"
Dynamic: license-file

# causal-toolkit-qingwei

[![Tests](https://github.com/qingweizhang054/causal-toolkit-qingwei-package/actions/workflows/tests.yml/badge.svg)](https://github.com/qingweizhang054/causal-toolkit-qingwei-package/actions)

[![PyPI version](https://badge.fury.io/py/causal-toolkit-qingwei.svg?)](https://pypi.org/project/causal-toolkit-qingwei/)

A Python package for causal inference methods including:

- ATE estimation
- IPW (Inverse Probability Weighting)
- Doubly Robust Estimation
- S-Learner
- T-Learner
- X-Learner
- Double Machine Learning

---

## Installation

```bash
git clone https://github.com/qingweizhang054/causal-toolkit-qingwei-package.git
cd causal-toolkit-qingwei-package

uv venv
.venv\Scripts\activate

uv pip install -e ".[dev]"
````

---

## Run Tests

```bash id="dxr5lu"
uv run pytest
```

---

## Example Usage

```python id="9a67pi"
from causal_toolkit_qingwei import (
    calculate_ate_ci,
    ipw,
    x_learner_discrete
)
```

---

## Author

Qingwei Zhang  
University of Chicago
