Brain_Tumor_Classification

2D-CNN for Brain Tumor Classification

Data is available at https://www.kaggle.com/navoneel/brain-mri-images-for-brain-tumor-detection

For training your own model call the function TRAINING() with parameters:- Boolean ( want to save the model or not), Train_loader, Number of EPOCHS

For testing the model run the TESTING() function with no parameters.