A machine learning approach for risk factors analysis and survival prediction of Heart Failure patients

In this study, we propose machine learning (ML) for risk factors analysis and survival prediction of Heart Failure (HF) patients using a survival dataset. Five supervised ML methods are applied to the dataset: Decision Tree (DT), Decision Tree Regressor (DTR), Random Forest (RF), XGBoost, and Gradient Boosting (GB) algorithms. We compare the applied algorithms’ performances based on accuracy, precision, recall, F-measure, and log loss value and show RF provides the highest accuracy of 97.78%. The analysis of the risk factors shows the most predictive features based on coefficients and feature importance. The top six risk factors for HF patients are serum creatinine (SC), age, ejection fraction (EF), platelets, creatinine phosphokinase (CPK), and SS (SS). Further analysis of these factors shows…