Model save

README.md

@@ -1,13 +1,26 @@
 ---
-base_model: /content/drive/MyDrive/Seizure_EEG_Research/ViT_Seizure_Detection
 tags:
-- image-classification
 - generated_from_trainer
 datasets:
 - arrow
+metrics:
+- matthews_correlation
 model-index:
 - name: ViT_Seizure_Detection
-  results: []
+  results:
+  - task:
+      name: Image Classification
+      type: image-classification
+    dataset:
+      name: arrow
+      type: arrow
+      config: default
+      split: test
+      args: default
+    metrics:
+    - name: Matthews Correlation
+      type: matthews_correlation
+      value: 0.4389188148304588
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -15,15 +28,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # ViT_Seizure_Detection
 
-This model is a fine-tuned version of [/content/drive/MyDrive/Seizure_EEG_Research/ViT_Seizure_Detection](https://huggingface.co//content/drive/MyDrive/Seizure_EEG_Research/ViT_Seizure_Detection) on the beans dataset.
+This model was trained from scratch on the arrow dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 0.3040
-- eval_matthews_correlation: 0.4261
-- eval_runtime: 1091.8559
-- eval_samples_per_second: 104.441
-- eval_steps_per_second: 13.056
-- epoch: 3.14
-- step: 40000
+- Loss: 0.3495
+- Matthews Correlation: 0.4389
 
 ## Model description
 
@@ -50,6 +58,17 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 4
 
+### Training results
+
+| Training Loss | Epoch | Step  | Validation Loss | Matthews Correlation |
+|:-------------:|:-----:|:-----:|:---------------:|:--------------------:|
+| 0.0742        | 0.79  | 10000 | 0.2080          | 0.4431               |
+| 0.0409        | 1.57  | 20000 | 0.2175          | 0.4470               |
+| 0.0345        | 2.36  | 30000 | 0.2514          | 0.4717               |
+| 0.0184        | 3.14  | 40000 | 0.3040          | 0.4261               |
+| 0.0092        | 3.93  | 50000 | 0.3495          | 0.4389               |
+
+
 ### Framework versions
 
 - Transformers 4.34.1

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:50cc017e8e32edbfa0d17c2fae7b56be5f356fa794a222ed5528897388452513
+oid sha256:2204d4d930925aec4b8d85b069448b6c796d2251f2c7150c1cdcd5e45c2c9a34
 size 343269162

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:94ed5a2455b5babf688e0fd413ecb6536e5a27992343df7896c171353305f839
+oid sha256:193070fc9050884640f8dbe8b5db5af0aaebc642ae39034da5a0e03c6e360a75
 size 4600