|
|
--- |
|
|
license: apache-2.0 |
|
|
language: |
|
|
- en |
|
|
metrics: |
|
|
- accuracy |
|
|
pipeline_tag: image-classification |
|
|
tags: |
|
|
- medical |
|
|
- biology |
|
|
library_name: keras |
|
|
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 |
