linakosmina commited on
Commit
d8b9b36
·
verified ·
1 Parent(s): ced1430

Update README.md

Browse files

Model Description.
This model is a fine-tuned EfficientNetB0 for classifying brain tumors from MRI scans. It was trained using transfer learning on a publicly available dataset from Kaggle.
The model is capable of distinguishing between four classes: pituitary, glioma, meningioma, and no_tumor.

Intended Use.
This model is intended for non-commercial and research purposes only. It can be used for:
- Classifying brain tumors in MRI images.
- Serving as a baseline for further research.

Performance.
The model was evaluated on a held-out test set and achieved the following results:
Metric Score
Accuracy 99.31%
Precision 99%
Recall 99%
F1-Score 99%

Training Data.
The model was trained on the Brain Tumor MRI Dataset available on Kaggle.
- Dataset Link: https://www.kaggle.com/datasets/masoudnickparvar/brain-tumor-mri-dataset

Training Details
- Base Model: EfficientNetB0 (pre-trained on ImageNet)
- Framework: TensorFlow / Keras
- Training Platform: Google Colaboratory

Files changed (1) hide show
  1. README.md +6 -3
README.md CHANGED
@@ -1,3 +1,6 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model:
4
+ - google/efficientnet-b0
5
+ pipeline_tag: image-classification
6
+ ---