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

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -1
README.md CHANGED
@@ -3,4 +3,31 @@ license: apache-2.0
3
  base_model:
4
  - google/efficientnet-b0
5
  pipeline_tag: image-classification
6
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  base_model:
4
  - google/efficientnet-b0
5
  pipeline_tag: image-classification
6
+ ---
7
+
8
+ Model Description.
9
+ 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.
10
+ The model is capable of distinguishing between four classes: pituitary, glioma, meningioma, and no_tumor.
11
+
12
+ Intended Use.
13
+ This model is intended for non-commercial and research purposes only. It can be used for:
14
+ - Classifying brain tumors in MRI images.
15
+ - Serving as a baseline for further research.
16
+
17
+ Performance.
18
+ The model was evaluated on a held-out test set and achieved the following results:
19
+
20
+ Metric Score
21
+ Accuracy 99.31%
22
+ Precision 99%
23
+ Recall 99%
24
+ F1-Score 99%
25
+
26
+ Training Data.
27
+ The model was trained on the Brain Tumor MRI Dataset available on Kaggle.
28
+ - Dataset Link: https://www.kaggle.com/datasets/masoudnickparvar/brain-tumor-mri-dataset
29
+
30
+ Training Details
31
+ - Base Model: EfficientNetB0 (pre-trained on ImageNet)
32
+ - Framework: TensorFlow / Keras
33
+ - Training Platform: Google Colaboratory