Machine learning for subgroup discovery under treatment effect