surajjoshi commited on
Commit
3c6403f
1 Parent(s): e9bdb6a

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -47
README.md CHANGED
@@ -2,38 +2,9 @@
2
  license: apache-2.0
3
  tags:
4
  - generated_from_trainer
5
- datasets:
6
- - imagefolder
7
- metrics:
8
- - accuracy
9
- - f1
10
- - recall
11
- - precision
12
  model-index:
13
  - name: swin-tiny-patch4-window7-224-finetuned-brainTumorData
14
- results:
15
- - task:
16
- name: Image Classification
17
- type: image-classification
18
- dataset:
19
- name: imagefolder
20
- type: imagefolder
21
- config: default
22
- split: train
23
- args: default
24
- metrics:
25
- - name: Accuracy
26
- type: accuracy
27
- value: 0.9977843426883308
28
- - name: F1
29
- type: f1
30
- value: 0.9984067976633033
31
- - name: Recall
32
- type: recall
33
- value: 0.9978768577494692
34
- - name: Precision
35
- type: precision
36
- value: 0.9989373007438895
37
  ---
38
 
39
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -41,13 +12,7 @@ should probably proofread and complete it, then remove this comment. -->
41
 
42
  # swin-tiny-patch4-window7-224-finetuned-brainTumorData
43
 
44
- This model is a fine-tuned version of [microsoft/swin-tiny-patch4-window7-224](https://huggingface.co/microsoft/swin-tiny-patch4-window7-224) on the imagefolder dataset.
45
- It achieves the following results on the evaluation set:
46
- - Loss: 0.0101
47
- - Accuracy: 0.9978
48
- - F1: 0.9984
49
- - Recall: 0.9979
50
- - Precision: 0.9989
51
 
52
  ## Model description
53
 
@@ -77,16 +42,6 @@ The following hyperparameters were used during training:
77
  - lr_scheduler_warmup_ratio: 0.1
78
  - num_epochs: 4
79
 
80
- ### Training results
81
-
82
- | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Recall | Precision |
83
- |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:------:|:---------:|
84
- | 0.1884 | 1.0 | 95 | 0.0706 | 0.9705 | 0.9787 | 0.9756 | 0.9818 |
85
- | 0.1134 | 2.0 | 190 | 0.0364 | 0.9889 | 0.9920 | 0.9883 | 0.9957 |
86
- | 0.1031 | 3.0 | 285 | 0.0116 | 0.9963 | 0.9973 | 0.9947 | 1.0 |
87
- | 0.0746 | 4.0 | 380 | 0.0101 | 0.9978 | 0.9984 | 0.9979 | 0.9989 |
88
-
89
-
90
  ### Framework versions
91
 
92
  - Transformers 4.23.1
 
2
  license: apache-2.0
3
  tags:
4
  - generated_from_trainer
 
 
 
 
 
 
 
5
  model-index:
6
  - name: swin-tiny-patch4-window7-224-finetuned-brainTumorData
7
+ results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  ---
9
 
10
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
12
 
13
  # swin-tiny-patch4-window7-224-finetuned-brainTumorData
14
 
15
+ This model is a fine-tuned version of [microsoft/swin-tiny-patch4-window7-224](https://huggingface.co/microsoft/swin-tiny-patch4-window7-224) on an unknown dataset.
 
 
 
 
 
 
16
 
17
  ## Model description
18
 
 
42
  - lr_scheduler_warmup_ratio: 0.1
43
  - num_epochs: 4
44
 
 
 
 
 
 
 
 
 
 
 
45
  ### Framework versions
46
 
47
  - Transformers 4.23.1