Predicting customer churn with AI