CoolFace
Modelpublic

shumailabatool/brain_tumor_detection

sourceHugging Faceapache-2.0updated 2y agoView on Hugging Face
0likes6downloads
Model Card

BrainTumorClassification

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 LossEpochStepValidation LossAccuracy
0.320813550.31620.9461
0.039222680.11770.9893
0.032632670.25960.9899
0.035142670.33980.9888
0.027252660.26360.9912

Framework

  • —Keras
  • —Pytorch
  • —TensorFlow
  • —Datasets