---
license: apache-2.0
language:
- en
metrics:
- accuracy
pipeline_tag: image-classification
tags:
- medical
- biology
library_name: timm
datasets:
- rocky93/BraTS_segmentation
---


# Brain_Tumor_Classification

This model is a fine-tuned version on the braTS20 dataset.
It achieves the following results on the evaluation set:
- Loss: 0.0272
- Accuracy: 0.9912


## Model description

More information needed

## Intended uses & limitations

More information needed

## Training and evaluation data

More information needed

## Training procedure
### Training hyperparameters

The following hyperparameters were used during training:
- learning_rate: 0.000001
- min_lr=0.000001
- num_epochs: 5
- optimizer: Adam  and epsilon=1e-06

### Training results

| Training Loss |Epoch| Step | Validation Loss| Accuracy | 
|:-------------:|:--:|:----:|:---------------:|:--------:|
| 0.3208        | 1  | 355  | 0.3162          | 0.9461   |
| 0.0392        | 2  | 268  | 0.1177          | 0.9893   | 
| 0.0326        | 3  | 267  | 0.2596          | 0.9899   | 
| 0.0351        | 4  | 267  | 0.3398          | 0.9888   | 
| 0.0272        | 5  | 266  | 0.2636          | 0.9912   |
### Framework 

- Keras 
- Pytorch 
- TensorFlow
- Datasets