Update README.md
Browse files
README.md
CHANGED
@@ -53,11 +53,12 @@ This must be used for classification of x-ray images of the brain to diagnose of
|
|
53 |
|
54 |
The model was fine-tuned in the dataset [Mahadih534/brain-tumor-dataset](https://huggingface.co/datasets/Mahadih534/brain-tumor-dataset) that contains 253 brain images. This dataset was originally created by Yousef Ghanem.
|
55 |
|
|
|
|
|
|
|
56 |
|
57 |
-
## Training procedure
|
58 |
-
The model was fine-tuned
|
59 |
|
60 |
-
### Training hyperparameters
|
61 |
|
62 |
The following hyperparameters were used during training:
|
63 |
- learning_rate: 0.0003
|
|
|
53 |
|
54 |
The model was fine-tuned in the dataset [Mahadih534/brain-tumor-dataset](https://huggingface.co/datasets/Mahadih534/brain-tumor-dataset) that contains 253 brain images. This dataset was originally created by Yousef Ghanem.
|
55 |
|
56 |
+
The original dataset was splitted into training and evaluation subsets, 80% for training and 20% for evaluation.
|
57 |
+
For robust framework evaluation, the evaluation subset is further split into two equal parts for validation and testing.
|
58 |
+
This results in three distinct datasets: training, validation, and testing
|
59 |
|
|
|
|
|
60 |
|
61 |
+
### Training procedure/hyperparameters
|
62 |
|
63 |
The following hyperparameters were used during training:
|
64 |
- learning_rate: 0.0003
|