Prediction of Breast Cancer Based on Penalized Logistic Regression
CSTR:
Author:
Affiliation:

1.School of Mathematics and Statistics, Chongqing Technology and Business University, Chongqing 400067, China;2.Chongqing Key Laboratory of Economic and Social Applied Statistics, Chongqing Technology and Business University, Chongqing 400067, China;3.Research Center for Economy of Upper Reaches of the Yangtse River, Chongqing Technology and Business University, Chongqing 400067, China

Clc Number:

TP181;R737.9

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    In this paper, we mainly apply the breast cancer data from University of Wisconsin System to predict breast cancer using penalized logistic regression. Firstly, the ten indicators related to breast cancer are selected as the predictor variables. Then, logistic regression, the LASSO penalized logistic regression, the L2 penalized logistic regression and the elastic net penalized logistic regression are used as the four classifiers. 75% of the data set is used as the training set to build models. Finally, 25% test set, a confusion matrix and a ROC curve are used to evaluate their prediction accuracy. The results show that the LASSO penalized logistic regression performs best, whose prediction accuracy reaches 97.18%. The prediction performance of the elastic net penalized logistic regression changes with the increase of α, especially when α=0.9, the corresponding prediction accuracy is 97.18%, as good as that of LASSO penalized logistic regression. The L2 penalized logistic regression ranks the third and logistic regression performs the worst in prediction performance. Therefore, for the diagnosis of breast tumors, doctors can apply the LASSO penalized logistic regression and the elastic net penalized logistic regression to improve the diagnostic accuracy.

    Reference
    Related
    Cited by
Get Citation

HU Xuemei, XIE Ying, JIANG Huifeng. Prediction of Breast Cancer Based on Penalized Logistic Regression[J].,2021,36(6):1237-1249.

Copy
Related Videos

Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:December 03,2020
  • Revised:May 26,2021
  • Adopted:
  • Online: November 25,2021
  • Published:
Article QR Code