causalml
causalml copied to clipboard
Add return_ci to learner.predict method
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)