uber / causalml

Uplift modeling and causal inference with machine learning algorithms

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

return_ci=True is not properly passed in get_ate_ci of the Sensitivity class in sensitivity.py

lee-junseok opened this issue · comments

Describe the bug
Some learner's estimate_ate function requires return_ci = True to return ate_lower, ate_upper together with ate. get_ate_ci of the Sensitivity class does not pass return_ci = True to estimate_ate.

To Reproduce
Steps to reproduce the behavior: Run sensitivity_analysis method with a class object Sensitivity(..., learner = XGBTRegressor(...), ...)

It gives
ValueError: not enough values to unpack (expected 3, got 1)

Expected behavior
Get sensitivity analysis result without a ValueError

Screenshots
image
Learner: XGBTRegressor()

Environment (please complete the following information):

  • OS: [Ubuntu (Jupyter notebook)]
  • Python Version: [3.7]
  • Versions of Major Dependencies (pandas, scikit-learn, cython): [pandas==1.35, scikit-learn==1.0, cython==0.29.34]

Additional context
None