bombshelll
/

swin-brain-modality-classification

Image Classification

Generated from Trainer

Model card Files Files and versions

bombshelll commited on Mar 2, 2025

Commit

d353fcf

·

verified ·

1 Parent(s): beb39d1

Model save

Files changed (1) hide show

README.md +5 -5

README.md CHANGED Viewed

@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
 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.
 It achieves the following results on the evaluation set:
-- Loss: 0.0349
-- Accuracy: 0.9821
 ## Model description
@@ -53,9 +53,9 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.7628        | 1.0   | 12   | 0.1582          | 0.9702   |
-| 0.1095        | 2.0   | 24   | 0.0339          | 0.9881   |
-| 0.0597        | 3.0   | 36   | 0.0349          | 0.9821   |
 ### Framework versions

 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.
 It achieves the following results on the evaluation set:
+- Loss: 0.0197
+- Accuracy: 0.9956
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 0.7116        | 1.0   | 16   | 0.1466          | 0.9333   |
+| 0.0955        | 2.0   | 32   | 0.0323          | 0.9911   |
+| 0.0464        | 3.0   | 48   | 0.0197          | 0.9956   |
 ### Framework versions