A Machine Learning-Based Framework for Diagnosis of Breast Cancer

Abstract

Machine learning is used in the health care sector due to its ability to make predictions. Nowadays major cause of death in women is due to breast cancer. In this paper, a machine learning-based framework for the diagnosis of breast cancer has been proposed. The authors have used different feature selection methods on Breast Cancer Wisconsin (Diagnostic) dataset i.e. Chi-square, Pearson correlation between features and Feature importance. The competency of the feature selection methods has been analyzed using different machine learning classifiers on different performance parameters like accuracy, sensitivity, specificity, precision, and F-measure. Random Forest (RF), Extra Tree Classifier (ETC), and Logistic Regression (LR) machine learning classifiers have been used by the authors. Results reveal that FI (Feature Importance) is the preeminent feature selection method among all others used when applied with different classifiers. Results also show that the ETC machine learning classifier gives the best accuracy result in comparison with RF and LR classifiers.