You are on page 1of 1

Link to dataset : https://figshare.

com/articles/dataset/brain_tumor_dataset/1512427

Train,val,test set split : The filename of images belonging to each split is given
in the train_data.csv,val_data.csv,test_data.csv respectively.

Confusion matrix of each split is also given as images

model_summary.txt : contains the summary of the model - the layer name, the output
shape of the layer, the number of parameters.

BRAIN_TUMOR - Train.ipynb - Notebook used to train the Model

Prediction - To use the model for Inference


***********************************************************************************
***********
EfficientNet B3 model was trained using the images from the COVID Radiography
dataset

Used tensorflow 2 to train the model.

Used NAdam (Nesterov Adam) optimizer to train the model. One Cycle Learning Policy
was used for training the model.

You might also like