causalml icon indicating copy to clipboard operation
causalml copied to clipboard

Add return_ci to learner.predict method

Open mathigatti opened this issue 4 years ago • 0 comments

Hi, would it be possible to get confidence intervals from a learner predict method?

Something like this

cate, cate_lb, cate_ub = learner.predict(X=X, treatment=treatment, y=y, return_ci=True)

mathigatti avatar Feb 18 '21 17:02 mathigatti