A two‐stage prediction model for heterogeneous effects of treatments. (27th May 2021)