Causal machine learning for predicting treatment outcomes