Skip to content

Here I tried various Machine Learning algorithms on different cancer's dataset present in CSV format.

License

Notifications You must be signed in to change notification settings

digamjain/Cancer-Cell-Prediction

Repository files navigation

Cancer-Cell-Prediction

Present here you can find various models specifically designed to curate to the various undermentioned datasets on various popular algorithms which are highly accepted on this type of data. My main objective was to use the various cancer related classification datasets that are publicly available and are found easily to predict cancer malignancy. In all you can find a comparative analysis of each algorithm with respect to the dataset used.

Datasets used

  • Breast Cancer Wisconsin (Diagnostic) Dataset
  • Lung Cancer Dataset
  • Prostate Cancer Dataset
  • Brain Tumour Dataset

image

Algorithms used

  • Logistic Regression
  • Decision Tree Classifier
  • Random Forest Classifier
  • Gaussian Naive Bayes

Results

Logistic Regression

image

Decision Tree Classifier

image

Random Forest Classifier

image

Gaussian Naive Bayes

image

Further Work

I am working on a model which further can classify and get better results than the above mentioned algorithms. The results which I have achieved so far are shown below.

image

In future I look forward to improve the major defects in my model and algorithm like reducing high time and space consumption that it faces for larger datasets and further improve the accuracy.