Using AI for survival predictions of heart failure patients