Using AI to predict heart attacks