File size: 1,358 Bytes
6bf654b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
---
license: apache-2.0
datasets: brain-tumor-image-dataset-semantic-segmentation
metrics:
- accuracy
- f1
- precision
- recall
pipeline_tag: image-classification
tags:
- brain-tumor
- image-classification
- keras
- tensorflow
- cnn
- mri
- healthcare
---
# Tumor Detection ML Model
## Model Description
This model is designed to classify brain tumor images using a Convolutional Neural Network (CNN). It has been trained and fine-tuned on a labeled dataset of brain tumor MRI images.
## Training Details
- **Framework:** TensorFlow/Keras
- **Optimizer:** Adam with a learning rate scheduler
- **Loss Function:** Categorical Crossentropy
- **Data Augmentation:** Includes rotation, width/height shift, zoom, and horizontal flipping.
- **Hyperparameter Tuning:** Performed using Keras Tuner.
## Metrics
The following metrics were used to evaluate the model's performance:
- **Accuracy:** Measures the overall correctness of predictions.
- **F1 Score:** Balances precision and recall.
- **Precision:** Indicates the proportion of true positives among positive predictions.
- **Recall:** Indicates the proportion of true positives among all actual positives.
## Usage
You can load the model using the Hugging Face Transformers library:
```python
from transformers import AutoModel
model = AutoModel.from_pretrained("YourUsername/Tumor_detection_ML_Model")
|