starpreeda commited on
Commit
7e50272
·
verified ·
1 Parent(s): f279129

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -7
README.md CHANGED
@@ -71,18 +71,21 @@ x = Dropout(0.4)(x)
71
  outputs = Dense(4, activation='softmax')(x)
72
 
73
  model = Model(inputs=base_model.input, outputs=outputs)
74
-
75
  # Load Weights
76
  model.load_weights("efficientnetb0_finetuned_brain_mri.keras")
77
 
 
 
 
 
 
 
 
 
 
78
 
79
- ## 📊 Dataset & Training Data
80
 
81
- - **Source:** Brain Tumor MRI Dataset (Kaggle)
82
- - **Data Distribution:**
83
- - **Training Set:** MRI images augmented with rotation, shift, zoom, and horizontal flips.
84
- - **Testing Set:** Independent brain MRI scans for evaluation.
85
- - **Classes:** 4 categories (Glioma, Meningioma, No Tumor, Pituitary).
86
 
87
  ### Confusion Matrix & Training Performance
88
  ![Confusion Matrix](https://huggingface.co/starpreeda/BrainTumorTest/resolve/main/Figure_1.png)
 
71
  outputs = Dense(4, activation='softmax')(x)
72
 
73
  model = Model(inputs=base_model.input, outputs=outputs)
 
74
  # Load Weights
75
  model.load_weights("efficientnetb0_finetuned_brain_mri.keras")
76
 
77
+ 📊 Dataset & Training Data
78
+
79
+ Source: Brain Tumor MRI Dataset (Kaggle)
80
+
81
+ Data Distribution:
82
+
83
+ Training Set: MRI images augmented with rotation, shift, zoom, and horizontal flips.
84
+
85
+ Testing Set: Independent brain MRI scans for evaluation.
86
 
87
+ Classes: 4 categories (Glioma, Meningioma, No Tumor, Pituitary).
88
 
 
 
 
 
 
89
 
90
  ### Confusion Matrix & Training Performance
91
  ![Confusion Matrix](https://huggingface.co/starpreeda/BrainTumorTest/resolve/main/Figure_1.png)