Heart diseases are among the primary causes of morbidity and mortality worldwide. Over the years, the incidence of these diseases has been increasing due to various factors contributing to their global prevalence, mainly resulting from lifestyle choices, misdiagnoses, or incorrect treatments due to late identification of these diseases. Artificial intelligence technologies, particularly machine learning algorithms, have gained significant momentum in the medical field by enabling the creation of models that predict diseases to facilitate timely diagnoses and treatments. The proposed model predicts cardiac diseases using a robust machine learning-based classification approach. The methodology consists of four phases: Dataset acquisition; Preprocessing (Label Encoding, One-Hot Encoding, SMOTE, and Standardization); Model implementation (SVM, LR, NB, KNN, DNN, MLP, and SGD); Hyperparameter tuning (GridSearchCV); and Model Evaluation. The best results, obtained with an 80% training and 20% testing partition, were achieved using the MLP algorithm. Its metrics surpassed those of the other models, with an Accuracy of 94%, Precision of 93%, Recall of 93%, F1-Score of 93%, and an AUC value of 0.981 for the ROC curve. In conclusion, the results demonstrate that a well-applied machine learning algorithm is highly effective for predicting cardiac diseases using patient medical data, which can aid in facilitating timely diagnoses and treatments.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Robust Heart Disease Classification Model Based on Machine Learning

  • Juan Torres,
  • Wilfredo Ticona

摘要

Heart diseases are among the primary causes of morbidity and mortality worldwide. Over the years, the incidence of these diseases has been increasing due to various factors contributing to their global prevalence, mainly resulting from lifestyle choices, misdiagnoses, or incorrect treatments due to late identification of these diseases. Artificial intelligence technologies, particularly machine learning algorithms, have gained significant momentum in the medical field by enabling the creation of models that predict diseases to facilitate timely diagnoses and treatments. The proposed model predicts cardiac diseases using a robust machine learning-based classification approach. The methodology consists of four phases: Dataset acquisition; Preprocessing (Label Encoding, One-Hot Encoding, SMOTE, and Standardization); Model implementation (SVM, LR, NB, KNN, DNN, MLP, and SGD); Hyperparameter tuning (GridSearchCV); and Model Evaluation. The best results, obtained with an 80% training and 20% testing partition, were achieved using the MLP algorithm. Its metrics surpassed those of the other models, with an Accuracy of 94%, Precision of 93%, Recall of 93%, F1-Score of 93%, and an AUC value of 0.981 for the ROC curve. In conclusion, the results demonstrate that a well-applied machine learning algorithm is highly effective for predicting cardiac diseases using patient medical data, which can aid in facilitating timely diagnoses and treatments.