Hugging Face's logo

shumailabatool
/
brain_tumor_detection

Image Classification
Keras
English
medical
biology
Model card Files Community
shumailabatool commited on
Commit
71389d7
·
verified ·
1 Parent(s): b8df552

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -10
README.md CHANGED
@@ -39,22 +39,18 @@ More information needed
39
  The following hyperparameters were used during training:
40
  - learning_rate: 0.000001
41
  - min_lr=0.000001
42
- - train_batch_size: 32
43
- - eval_batch_size: 32
44
- - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
- - lr_scheduler_type: linear
46
- - lr_scheduler_warmup_ratio: 0.1
47
  - num_epochs: 5
 
48
 
49
  ### Training results
50
 
51
  | Training Loss |Epoch| Step | Validation Loss| Accuracy |
52
  |:-------------:|:--:|:----:|:---------------:|:--------:|
53
- | 0.3208 | 1 | 83 | 0.3162 | 0.9461 |
54
- | 0.0392 | 2 | 166 | 0.1177 | 0.9893 |
55
- | 0.0326 | 3 | 249 | 0.2596 | 0.9899 |
56
- | 0.0351 | 4 | 332 | 0.3398 | 0.9888 |
57
- | 0.0272 | 5 | 415 | 0.2636 | 0.9912 |
58
  ### Framework
59
 
60
  - Keras
 
39
  The following hyperparameters were used during training:
40
  - learning_rate: 0.000001
41
  - min_lr=0.000001
 
 
 
 
 
42
  - num_epochs: 5
43
+ - optimizer: Adam and epsilon=1e-06
44
 
45
  ### Training results
46
 
47
  | Training Loss |Epoch| Step | Validation Loss| Accuracy |
48
  |:-------------:|:--:|:----:|:---------------:|:--------:|
49
+ | 0.3208 | 1 | 355 | 0.3162 | 0.9461 |
50
+ | 0.0392 | 2 | 268 | 0.1177 | 0.9893 |
51
+ | 0.0326 | 3 | 267 | 0.2596 | 0.9899 |
52
+ | 0.0351 | 4 | 267 | 0.3398 | 0.9888 |
53
+ | 0.0272 | 5 | 266 | 0.2636 | 0.9912 |
54
  ### Framework
55
 
56
  - Keras