Hyperparameter tuning and comparison of k nearest neighbour and decision tree algorithms for cardiovascular disease prediction Online publication date: Fri, 29-Oct-2021
by Preeti Bhowmick; Sachin Gajjar; Shital Chaudhary
International Journal of Swarm Intelligence (IJSI), Vol. 6, No. 2, 2021
Abstract: This work aims to do hyperparameter tuning and comparison of k nearest neighbour (kNN) and decision tree algorithms for cardiovascular disease (CVD) prediction using Framingham dataset. Hyperparameter tuning is done to find optimal value of k using Euclidean, Manhattan and Chebyshev distance metric in kNN. Hyperparameter tuning is done in decision tree, to find optimal value of the depth of the tree using Gini index and information gain attribute selection method. The algorithms are compared on the basis of confusion matrix, accuracy, error rate, specificity, recall, precision, F1 score, execution time and ROC-AUC. The results show the accuracy of the decision tree is 2% less than kNN but decision tree is 46.36% more time efficient. The AUC value of kNN is 0.613 and decision tree is 0.588. Decision tree is more appropriate for predicting CVD, as it predicted ten more true positives in confusion matrix.
Online publication date: Fri, 29-Oct-2021
Go to Inderscience Online Journals to access the Full Text of this article.
If you are not a subscriber and you just want to read the full contents of this article, buy online access here.Complimentary Subscribers, Editors or Members of the Editorial Board of the International Journal of Swarm Intelligence (IJSI):
Login with your Inderscience username and password:
Want to subscribe?
A subscription gives you complete access to all articles in the current issue, as well as to all articles in the previous three years (where applicable). See our Orders page to subscribe.
If you still need assistance, please email firstname.lastname@example.org