A case study of microarray breast cancer classification using machine learning algorithms with grid search cross validation

Citation

Mohd Ali, Nursabillilah and Besar, Rosli and Ab Aziz, Nor Azlina (2023) A case study of microarray breast cancer classification using machine learning algorithms with grid search cross validation. Bulletin of Electrical Engineering and Informatics, 12 (2). pp. 1047-1054. ISSN 2089-3191

[img] Text
91.pdf - Published Version
Restricted to Repository staff only

Download (428kB)

Abstract

Breast cancer is one of the leading causes of death and most frequently diagnosed cancer amongst women. Annually, almost half a million women do not survive the disease and die from breast cancer. Machine learning is a subfield of artificial intelligence (AI) and computer science that uses data and algorithms to mimic how humans learn, and gradually improving its accuracy. In this work, simple machine learning methods are used to classify breast cancer microarray data to normal and relapse. The data is from the gene expression omnibus (GEO) website namely GSE45255 and GSE15852. These two datasets are integrated and combined to form a single dataset. The study involved three machine learning algorithms, random forest (RF), extra tree (ET), and support vector machine (SVM). Grid search cross validation (CV) is applied for hyperparameter tuning of the algorithms. The result shows that the tuned SVM is best among the tested algorithms with accuracy of 97.78%. In the future it is recommended to include feature selection method to get the optimal features and better classification accuracies.

Item Type: Article
Uncontrolled Keywords: Breast Cancer, Classification, Grid SearchCV, Microarray
Subjects: Q Science > QP Physiology
Divisions: Faculty of Engineering and Technology (FET)
Depositing User: Ms Nurul Iqtiani Ahmad
Date Deposited: 31 Jan 2023 06:30
Last Modified: 31 Jan 2023 06:30
URII: http://shdl.mmu.edu.my/id/eprint/11105

Downloads

Downloads per month over past year

View ItemEdit (login required)