Expert cancer model using supervised algorithms with a LASSO selection approach

Pronab Ghosh, Asif Karim, Syeda Tanjila Atik, Saima Afrin, Mohd. Saifuzzaman

Abstract


One of the most critical issues of the mortality rate in the medical field in current times is breast cancer. Nowadays, a large number of men and women is facing cancer-related deaths due to the lack of early diagnosis systems and proper treatment per year. To tackle the issue, various data mining approaches have been analyzed to build an effective model that helps to identify the different stages of deadly cancers. The study successfully proposes an early cancer disease model based on five different supervised algorithms such as Logistic Regression (henceforth LR), Decision Tree (henceforth DT), Random Forest (henceforth RF), Support Vector Machine (henceforth SVM), and K-Nearest Neighbor (henceforth KNN). After an appropriate preprocessing of the dataset, Least Absolute Shrinkage and Selection Operator (LASSO) was used for feature selection (FS) using a 10-fold cross-validation (CV) approach. Employing LASSO with 10-fold cross-validation has been a novel steps introduced in this research. Afterwards, different performance evaluation metrics were measured to show accurate predictions based on the proposed algorithms. The result indicated top accuracy was received from RF classifier, approximately 99.41% with the integration of LASSO. Finally, a comprehensive comparison was carried out on Wisconsin Breast Cancer (Diagnostic) dataset (WBCD) together with some current works containing all features.

Keywords


breast cancer; decision tree; SVM;



DOI: http://doi.org/10.11591/ijece.v11i3.pp%25p
Total views : 0 times


Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.

ISSN 2088-8708, e-ISSN 2722-2578