Comparative Performance Analysis of Different Classification Algorithm for the Purpose of Prediction of Lung Cancer

  • Subrato BharatiEmail author
  • Prajoy Podder
  • Rajib Mondal
  • Atiq Mahmood
  • Md. Raihan-Al-Masud
Conference paper
Part of the Advances in Intelligent Systems and Computing book series (AISC, volume 941)


At present, Lung cancer is the serious and number one cause of cancer deaths in both men and women in worldwide. Cigarette Smoking can be considered as the principle cause for lung cancer. It can arise in any portion of the lung, but the lung cancer 90%–95% are thought to arise from the epithelial cells, this cells lining the bigger and smaller airways (bronchi and bronchioles). Mainly this paper focus on diagnosing the lung cancer disease using various classification algorithm with the help of python based data mining tools. For this purpose, Lung Cancer dataset has been collected from UCI machine learning repository. Three types of pathological cancers have been illustrated in the datasets. In this research paper, the proficiency and potentiality of the classification of Naïve Bayes, Logistic Regression, K-Nearest Neighbors (KNN), Tree, Random Forest, Neural Network in examining the Lung cancer dataset has been investigated to predict the presence of lung cancer with highest accuracy. Performance of the classification algorithms has been compared in terms of classification accuracy, precision, recall, F1 score. Finding out the confusion matrix, Classifier’s overall accuracy, user and producer accuracy individually for each classes and value of kappa statistics have been determined in this paper. Area under Receiver Operating Characteristic (ROC) curve and distribution plot of the mentioned classifiers have also been showed in this paper. This paper also implemented Principal component analysis (PCA) and visualized classification tree, Multidimensional scaling (MDS) and Hierarchical Clustering.


Naïve Bayes Logistic regression Decision tree Random forest Neural network AUC KNN Confusion matrix PCA MDS Hierarchical clustering 


  1. 1.
    Alberg, A.J., Brock, M.V., Samet, J.M.: Epidemiology of lung cancer. In: Murray & Nadel’s Textbook of Respiratory Medicine, 6th edn., Chap. 52. Saunders Elsevier (2016)Google Scholar
  2. 2.
    Thun, M.J., Hannan, L.M., Adams-Campbell, L.L., et al.: Lung cancer occurrence in never-smokers: an analysis of 13 cohorts and 22 cancer registry studies. PLoS Med. 5(9), e185 (2008)CrossRefGoogle Scholar
  3. 3.
    Hong, Z.Q., Yang, J.Y.: Optimal discriminant plane for a small number of samples and design method of classifier on the plane. Pattern Recogn. 24(4), 317–324 (1991)MathSciNetCrossRefGoogle Scholar
  4. 4.
    Oh, J.H., Al-Lozi, R., El Naqa, I.: Application of machine learning techniques for prediction of radiation pneumonitis in lung cancer patients. In: 8th International Conference on Machine Learning and Applications, ICMLA 2009, pp. 478–483 (2009)Google Scholar
  5. 5.
    Lynch, C.M., Abdollahi, B., Fuqua, J.D., de Carlo, A.R., et al.: Prediction of lung cancer patient survival via supervised machine learning classification techniques. Int. J. Med. Inform. 108, 1–8 (2017)CrossRefGoogle Scholar
  6. 6.
    Tazin, N., Sabab, S.A., Chowdhury, M.T.: Diagnosis of chronic kidney disease using effective classification and feature selection technique. In: International Conference on Medical Engineering, Health Informatics and Technology (MediTec) (2016)Google Scholar
  7. 7.
    Kirubha, V., Manju Priya, S.: Comparison of classification algorithms in lung cancer risk factor analysis. Int. J. Sci. Res. (IJSR) 6(2), 1794–1797 (2017)Google Scholar
  8. 8.
    Abdar, M., Kalhori, S.R.N., Sutikno, T., Subroto, I.M.I., Arji, G.: Comparing performance of data mining algorithms in prediction heart diseases. Int. J. Electr. Comput. Eng. (IJECE) 5(6), 1569–1576 (2015)Google Scholar
  9. 9.
    Hristea, F.T.: The Naïve Bayes Model for Unsupervised Word Sense Disambiguation: Aspects Concerning Feature Selection. Springer, Berlin (2012)Google Scholar
  10. 10.
    Hilbe, J.M.: Logistic Regression Models. CRC Press, Boca Raton (2009)CrossRefGoogle Scholar
  11. 11.
    Retmin Raj, C.S., Nehemiah, H.K., Elizabeth, D.S., Kannan, A.: A novel feature-significance based k-nearest neighbour classification approach for computer aided diagnosis of lung disorders. Curr. Med. Imaging Rev. 14(2), 289–300(12) (2018)CrossRefGoogle Scholar
  12. 12.
    Kamiński, B., Jakubczyk, M., Szufel, P.: A framework for sensitivity analysis of decision trees. Central Eur. J. Oper. Res. 26, 135–159 (2017)MathSciNetCrossRefGoogle Scholar
  13. 13.
    Trevor, H., Robert, T., Jerome, F.: The Elements of Statistical Learning, 2nd edn. Springer, Berlin (2008)zbMATHGoogle Scholar
  14. 14.
    Tosh, C.R., Ruxton, G.D.: Modelling Perception with Artificial Neural Networks. Cambridge University Press, Cambridge (2010)CrossRefGoogle Scholar

Copyright information

© Springer Nature Switzerland AG 2020

Authors and Affiliations

  • Subrato Bharati
    • 1
    Email author
  • Prajoy Podder
    • 1
  • Rajib Mondal
    • 1
  • Atiq Mahmood
    • 2
  • Md. Raihan-Al-Masud
    • 3
  1. 1.Department of EEERanada Prasad Shaha UniversityNarayanganjBangladesh
  2. 2.Department of EEEWorld University of BangladeshDhakaBangladesh
  3. 3.Department of EEEStamford University BangladeshDhakaBangladesh

Personalised recommendations