Skip to main content
Join the official 2019 Python Developers SurveyStart the survey!

A library of causal inference tools by IBM Haifa Research Labs

Project description

Build Status Test Coverage PyPI version Documentation Status

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


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for causallib, version 0.5.0b0
Filename, size File type Python version Upload date Hashes
Filename, size causallib-0.5.0b0-py3-none-any.whl (123.9 kB) File type Wheel Python version py3 Upload date Hashes View hashes
Filename, size causallib-0.5.0b0.tar.gz (92.6 kB) File type Source Python version None Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN SignalFx SignalFx Supporter DigiCert DigiCert EV certificate StatusPage StatusPage Status page