Instructions to use starpreeda/BrainTumorTest with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Keras
How to use starpreeda/BrainTumorTest with Keras:
# Available backend options are: "jax", "torch", "tensorflow". import os os.environ["KERAS_BACKEND"] = "jax" import keras model = keras.saving.load_model("hf://starpreeda/BrainTumorTest") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files## 📈 Evaluation Results
Model performance evaluated on the independent testing dataset:
- **Overall Test Accuracy:** `~95-98%`
- **Classification Performance:**
<p align="center">
<img src="Figure_1.png" alt="Confusion Matrix" width="45%">
<img src="Figure_2.png" alt="Training Graphs" width="45%">
</p>
**Note:** Performance metrics were calculated using `sklearn.metrics.classification_report` after fine-tuning the top 40 layers of EfficientNetB0.
README.md
CHANGED
|
@@ -73,4 +73,17 @@ outputs = Dense(4, activation='softmax')(x)
|
|
| 73 |
model = Model(inputs=base_model.input, outputs=outputs)
|
| 74 |
|
| 75 |
# Load Weights
|
| 76 |
-
model.load_weights("efficientnetb0_finetuned_brain_mri.keras")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
|
| 88 |
+
|
| 89 |
+
|