Heart Disease Dataset Clusterization

Stud Health Technol Inform. 2019:261:162-167.

Abstract

Clusterization is a promising group of methods in the context of patient similarity. However, results of clustering are not often clear for physicians as well as different clustering methods can produce different results. We have examined a well-known dataset and implemented 3 clustering methods (k-means, Agglomerative and Spectral). We have compared and evaluated clusters and their correlation with data attributes. In contrast to original dataset's target value, the clusters correlated with only a few attributes. Finally, we train 2 predictive models based on k-nearest neighbors (KNN) algorithm and Artificial Neural Network (ANN). Models evaluation demonstrates that using the results of clustering algorithms as predictive attribute give a higher F-score than the original target attribute.

Keywords: Patient similarity; clusterization; patient classification.

MeSH terms

  • Algorithms*
  • Cluster Analysis
  • Data Analysis*
  • Heart Diseases*
  • Humans
  • Neural Networks, Computer