Update README.md
Browse files
README.md
CHANGED
@@ -36,15 +36,18 @@ This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https:
|
|
36 |
It achieves the following results on the evaluation set:
|
37 |
- Loss: 0.0014
|
38 |
- Accuracy: 1.0
|
39 |
-
-
|
40 |
-
-
|
41 |
-
-
|
42 |
-
-
|
43 |
-
-
|
44 |
-
-
|
45 |
-
-
|
46 |
-
-
|
47 |
-
-
|
|
|
|
|
|
|
48 |
|
49 |
## Model description
|
50 |
|
@@ -79,7 +82,7 @@ The following hyperparameters were used during training:
|
|
79 |
|
80 |
### Training results
|
81 |
|
82 |
-
| Training Loss | Epoch | Step | Validation Loss | Accuracy | Weighted
|
83 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|:-----------:|:--------:|:--------:|:---------------:|:------------:|:------------:|:------------------:|:---------------:|:---------------:|
|
84 |
| 0.2079 | 1.0 | 938 | 0.0193 | 0.996 | 0.9960 | 0.996 | 0.9960 | 0.996 | 0.996 | 0.9960 | 0.9960 | 0.996 | 0.9960 |
|
85 |
| 0.0154 | 2.0 | 1876 | 0.0068 | 0.9987 | 0.9987 | 0.9987 | 0.9987 | 0.9987 | 0.9987 | 0.9987 | 0.9987 | 0.9987 | 0.9987 |
|
|
|
36 |
It achieves the following results on the evaluation set:
|
37 |
- Loss: 0.0014
|
38 |
- Accuracy: 1.0
|
39 |
+
- F1
|
40 |
+
- Weighted: 1.0
|
41 |
+
- Micro: 1.0
|
42 |
+
- Macro: 1.0
|
43 |
+
- Recall
|
44 |
+
- Weighted: 1.0
|
45 |
+
- Micro: 1.0
|
46 |
+
- Macro: 1.0
|
47 |
+
- Precision
|
48 |
+
- Weighted: 1.0
|
49 |
+
- Micro: 1.0
|
50 |
+
- Macro: 1.0
|
51 |
|
52 |
## Model description
|
53 |
|
|
|
82 |
|
83 |
### Training results
|
84 |
|
85 |
+
| Training Loss | Epoch | Step | Validation Loss | Accuracy | Weighted F1 | Micro F1 | Macro F1 | Weighted Recall | Micro Recall | Macro Recall | Weighted Precision | Micro Precision | Macro Precision |
|
86 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|:-----------:|:--------:|:--------:|:---------------:|:------------:|:------------:|:------------------:|:---------------:|:---------------:|
|
87 |
| 0.2079 | 1.0 | 938 | 0.0193 | 0.996 | 0.9960 | 0.996 | 0.9960 | 0.996 | 0.996 | 0.9960 | 0.9960 | 0.996 | 0.9960 |
|
88 |
| 0.0154 | 2.0 | 1876 | 0.0068 | 0.9987 | 0.9987 | 0.9987 | 0.9987 | 0.9987 | 0.9987 | 0.9987 | 0.9987 | 0.9987 | 0.9987 |
|