Views
No views yet
| Artifact | Description |
|---|---|
models/dr_real.joblib | DR-Learner trained on real treatment |
models/cf_real.joblib | Causal Forest trained on real treatment |
models/x_real.joblib | X-Learner trained on real treatment (Best) |
models/dr_placebo.joblib | DR-Learner placebo baseline |
models/cf_placebo.joblib | Causal Forest placebo baseline |
models/x_placebo.joblib | X-Learner placebo baseline |
models/propensity_model.joblib | Propensity score model |
models/tau_dr_real.npy | Predicted CATE array — DR-Learner |
models/tau_cf_real.npy | Predicted CATE array — Causal Forest |
models/tau_x_real.npy | Predicted CATE array — X-Learner |
1from huggingface_hub import hf_hub_download
2import joblib, numpy as np
3
4# Load best model
5x_model = joblib.load(hf_hub_download("nguyennghia0902/uplift-criteo-kdd-lastupdate", "models/x_real.joblib"))
6
7# Predict CATE
8tau = x_model.effect(X_test_fe)