Skip to main content

deepCausal

deepCausal is a Python library for causal feature ranking and selection based on residual treatment effect estimation.
It uses double machine learning and residualization techniques to estimate the causal impact (ATE) of each feature on a continuous or categorical outcome.

🔍 Key Features

  • Supports both continuous and categorical outcomes
  • Uses residual-on-residual regression for causal interpretation
  • Identifies confounders automatically
  • Ranks features by causal strength, not just correlation

🚀 Usage

from deepCausal import DeepCausal import pandas as pd import numpy as np

df = pd.DataFrame({ 'X1': np.random.randn(100), 'X2': np.random.rand(100), 'X3': np.random.randn(100), 'Y': np.random.randn(100) })

X = df[['X1', 'X2', 'X3']] y = df['Y']

model = DeepCausal() model.fit(X, y, outcome_type='continuous')

print(model.get_feature_ate())

📄 License This project is licensed under the MIT License.

🧪 Disclaimer This method is experimental and currently under research evaluation. Feedback and pull requests are welcome!

Release files for causal-drift 0.1.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for causal-drift 0.1.0
File Size Uploaded
causal_drift-0.1.0.tar.gz 3.8 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for causal-drift 0.1.0
File Interpreter ABI Platform
causal_drift-0.1.0-py3-none-any.whl Python 3 none any Details

Total release size: 7.8 kB

Release files / causal_drift-0.1.0.tar.gz

Download URL causal_drift-0.1.0.tar.gz
Size 3.8 kB
Tags Source
SHA-256 checksum
How to use checksums
2519f32763d22488dcbd7953d1c888c1db51389a0e382e361d8163b0c0d0df07
BLAKE2b-256 checksum
How to use checksums
a57b6ae72be91f1f0b50c37476843e368259c96894d2a3d6c5a608ea1f76c587
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.1.0 CPython/3.13.5

Release files / causal_drift-0.1.0-py3-none-any.whl

Download URL causal_drift-0.1.0-py3-none-any.whl
Size 4.0 kB
Tags Python 3
SHA-256 checksum
How to use checksums
1d0cc9a3905279f1e4d4c7c40b1c2b40280f1a3f94eab78419b4507ac20fefec
BLAKE2b-256 checksum
How to use checksums
3bad3c4c974c16eec43b784a782f5f0222d1b75757b2ed8550e8a883e5b7cd8f
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.1.0 CPython/3.13.5

Release history Release notifications | RSS feed

This release

0.1.0 This release

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page