Visualization of Data Mining Techniques for the Prediction of Breast Cancer with High Accuracy Rates
- 1 Vellore Institute of Technology, India
- 2 Brighton and Sussex Medical School, United Kingdom
Copyright: © 2020 Vasudev Sharma, Raj Kumar Rajasekaran and Shreya Badhrinarayanan. This is an open access article distributed under the terms of the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original author and source are credited.
Breast cancer is one of the leading causes of death in women worldwide. Around one in 30 women are affected by breast cancer. Mammography has helped in detecting breast cancer in the early stages which have reduced mortality. The diagnosis of breast cancer is dependent on a variety of parameters. In this paper, we aim to create the best model for predicting breast cancer through preprocessing, feature extraction, data visualization and prediction using breast cancer data. Various visualization techniques like violin plot, grid plot, swarm plot and heat plot were utilized for proper feature extraction which has improved the accuracy of our results. For the purpose of prediction, we have used algorithms like the random forest, decision tree with single and multiple predictors, along with the commonly used statistical model, logistic regression model. We have also relied on 5-fold cross-validation methods to measure the unbiasedness of the prediction models for performance reasons. An analysis of the models was carried out and the best model was selected based on its accuracy. The results showcased that the random forest model provided an accuracy rate of 94.724% with decent 5-fold cross-validation, followed by the decision tree model which had an accuracy rate of 100% with poor 5-fold cross-validation. This was followed by the logistic regression model which had an accuracy rate of 88.442% with a low 5-fold cross-validation score.
- 1,387 Views
- 966 Downloads
- 2 Citations
- Data Visualization
- Violin Plot
- Swarm Plot
- Random Forest
- Logistic Regression
- Decision Tree
- 5-Fold Cross Validation