XGboost model to predict heart attack risk