resolves change requests in #166 (#701)

* adding new data generation function based on logistic regression for classification * adding new data generation function: modify docstring * adding example notebook for DGP * simplify function selection * use np.clip * remove Y1 * rm commented code block * rm rcoef, use 0.5 * switch to expit * switch to logit * remove blank lines * link with black --------- Co-authored-by: Zhenyu Zhao <[email protected]> Co-authored-by: Roland Stevenson <[email protected]>
uber · Nov 8, 2023 · 176cc78 · 176cc78
1 parent 9cd8ada
commit 176cc78
Show file tree

Hide file tree

Showing 3 changed files with 971 additions and 0 deletions.
diff --git a/causalml/dataset/__init__.py b/causalml/dataset/__init__.py
@@ -5,6 +5,7 @@
 from .regression import simulate_unrelated_treatment_control
 from .regression import simulate_hidden_confounder
 from .classification import make_uplift_classification
+from .classification import make_uplift_classification_logistic
 
 from .synthetic import get_synthetic_preds, get_synthetic_preds_holdout
 from .synthetic import get_synthetic_summary, get_synthetic_summary_holdout