Citation

Bau, Yoon Teck and Sasidaran, Tishanthini and Goh, Chien Le (2022) Improving Machine Learning Algorithms for Breast Cancer Prediction. Journal of System and Management Sciences, 12 (4). pp. 251-266. ISSN 1816-6075, 1818-0523

Text Vol.12.No.04.16.pdf - Published Version Restricted to Repository staff only Download (383kB)

Abstract

Early prediction of breast cancer can prevent death or receiving late treatment. The purpose of this research is to improve machine learning algorithms in predicting breast cancer that will assist patients and healthcare systems. The machine learning algorithms for the prediction of breast cancer are the methods applied in this research by using these following algorithms which are decision tree, random forest, naive Bayes, and gradient boosting due to their high performance. This research uses data from the breast cancer of Wisconsin (diagnostic) dataset of the general surgery department. The results from this research are that by using the stratified k-fold cross validation as a part of the random forest classifier achieved 100% for all four performance scores which are accuracy, recall, precision and F1. The stratified k-fold also improved two machine learning algorithms. In addition, data visualization was applied to the random forest algorithm for result understanding. The implication from the best method is that it could increase the number of accurate breast cancer detections. The values by selecting the best method from this research could assist doctors in early breast cancer detection and increase the number of breast cancer survival rates by receiving early treatment from accurate prediction.

Item Type:	Article
Uncontrolled Keywords:	Machine learning algorithms, decision tree, random forest, naive bayes, gradient boosting, classifier, breast cancer prediction, stratified k-fold, cross validation
Subjects:	Q Science > QR Microbiology > QR180 Immunology
Divisions:	Faculty of Computing and Informatics (FCI)
Depositing User:	Ms Nurul Iqtiani Ahmad
Date Deposited:	11 Oct 2022 05:50
Last Modified:	11 Oct 2022 05:50
URII:	http://shdl.mmu.edu.my/id/eprint/10530

Downloads

Edit (login required)