Comparison of Tree Based Model Feature Importances With SHAP Scores
Heart failure, which affects over 64 million individuals globally, is a serious health problem. Early and accurate prediction is crucial for reducing its impact. Researchers have employed various machine learning models to address this challenge. Our aim is to find whether the importance scores generated by random forest, gradient boost, and XGBoost agree with the explain-ability method. This paper shows the prediction model whose feature importances agree with the Shapley values for our selected dataset. We also compare feature importance scores of three prediction algorithms random forest, gradient boost, and XGBoost with SHAP (Shapley Additive exPlanations) to find their agreement on scores. We use the UCI “Heart Failure Clinical Records” collection from the UCI Machine Learning Repository, which has $\mathbf{2 9 9}$ patients in its heart failure dataset.
Authors
- Gaurav Parashar (ORCID: https://orcid.org/0000-0003-4869-1819)
- Nikhil Sharma (ORCID: https://orcid.org/0009-0002-6286-5037)
Publication Details
- Published
- 2024-11-21
- DOI
- https://doi.org/10.1109/akgec62572.2024.10868174
- Primary Topic
- Machine Learning and Data Classification
- Type
- article
- Field-Weighted Citation Impact
- 0.00