A library of causal inference tools by IBM Haifa Research Labs
Project description
IBM Causal Inference Library
A Python package for computational inference of causal effect.
Description
Causal inference analysis allows estimating of the effect of intervention on some outcome from observational data. It deals with the selection bias that is inherent to such data.
This python package allows creating modular causal inference models that internally utilize machine learning models of choice, and can estimate either individual or average outcome given an intervention. The package also provides the means to evaluate the performance of the machine learning models and their predictions.
The machine learning models must comply with scikit-learn's api
and contain fit()
and predict()
functions.
Categorical models must also implement predict_proba()
.
Installation
pip install causallib
Usage
In general, the package is imported using the name causallib
.
For example, use
from sklearn.linear_model import LogisticRegression
from causallib.estimation import IPW
ipw = IPW(LogisticRegression())
Comprehensive Jupyter Notebooks examples can be found in the examples directory.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for causallib-0.5.0b0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 94244d5d60c9437db6b3ce55ded40085a8d59bd67bae3fc1ce290e2c43407794 |
|
MD5 | b381c4931fa3ca9af9a451b9253cea55 |
|
BLAKE2b-256 | 6cc9bf502a2dd066bca053670f0a51356da2cc75ad7b5ac09bd09b2fb00af3fb |