Breast Cancer Classification

Installation

It is assumed that python is installed in your system.

pip3/pip install keras 2.4.3
pip3/pip install tensorflow 2.3.0
pip3/pip install numpy 1.18.5
pip3/pip install sklearn 0.23.2
pip3/pip install matplotlib 3.3.0

Description

• The dataset can be available at Kaggle.
• This model aims to classify the images of the people who have breast cancer which in turns helps to prevent it.

Method

• In the above python file, you will find 4 models which I have trained and measured their accuracies.
• From the 4 models one of them is pretrained vgg network, just for the sake of checking how it would perform on vgg16 model.
• The other 3 models were developed and you can find them in the same python file.
• The same python file also contains the code for building the dataset directory and please follow the path '/breast_cancer/datasets/original/' and keep the downloaded kaggle dataset file there otherwise it won't work.

Result

• pre trained vgg 16 model:  training accuracy: 82.96%   validation accuracy: 83.62%  testing accuracy: 83.61%

• self trained model_1:      training accuracy: 88.69%   validation accuracy: 88.84%  testing accuracy: ~88%

• self trained model_4:      training accuracy: 87%      validation accuracy: 87%     testing accuracy: 86%

• self trained model_2:      training accuracy: 88.67%   validation accuracy: 87.80%  testing accuracy: 87.66%

Output