Skip to content

Tools for analyzing and quantifying effects of confounder variables on machine learning model predictions.

License

Notifications You must be signed in to change notification settings

pni-lab/mlconfound

Repository files navigation

mlconfound

GitHub license GitHub release GitHub CI Documentation Status arXiv GitHub issues GitHub issues-closed Binder

Tools for analyzing and quantifying effects of counfounder variables on machine learning model predictions.

Install

pip install mlconfound

Usage

# y   : prediction target
# yhat: prediction
# c   : confounder

from mlconfound.stats import partial_confound_test

partial_confound_test(y, yhat, c)

Run the quickstart notebook in Binder: Binder

Read the docs for more details.

Documentation Documentation Status

https://mlconfound.readthedocs.io

Citation

T. Spisak, “Statistical quantification of confounding bias in machine learning models.” GigaScience 11 (2022) https://doi.org/10.1093/gigascience/giac082. Preprint on arXiv:2111.00814, 2021.