Predicting erectile dysfunction after treatment for localized prostate cancer