计算机科学
疾病
人工智能
机器学习
人机交互
数据科学
医学
病理
作者
Gangani Dharmarathne,Madhusha Bogahawaththa,Upaka Rathnayake,D.P.P. Meddage
标识
DOI:10.1016/j.iswa.2024.200428
摘要
Conventional machine learning techniques in diagnosing cardiovascular disease have a limitation owing to the lack of interpretability of models. This study utilised an explainable machine learning approach to predict the likelihood of having CVD. Four machine learning models were employed for CVD diagnosis: Decision Tree (DT), K-Nearest Neighbor (KNN), Random Forest (RF), and Extreme Gradient Boost (XGB). Shapley Additive Explanations (SHAP) were used to provide reasoning for the models' predictions. Using these models and explanations, a user interface was developed to assist in diagnosing CVD. All four classification models demonstrated good accuracy in diagnosing CVD, with the KNN model showcasing the best performance (Accuracy: 71 %). SHAP provided the reasoning behind KNN predictions, and the predictive interface was developed by embedding these explanations to provide transparency behind the model's decisions.
科研通智能强力驱动
Strongly Powered by AbleSci AI