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

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 2302-9285

Text 4838-13012-1-PB.PDF Download (429kB)

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:	Microarray, Breast cancer, Grid searchCV, Classification
Divisions:	Faculty of Electrical Engineering
Depositing User:	mr eiisaa ahyead
Date Deposited:	28 Feb 2023 08:03
Last Modified:	28 Feb 2023 08:03
URI:	http://eprints.utem.edu.my/id/eprint/26481
Statistic Details:	View Download Statistic

Actions (login required)

View Item