VIT_SEIZURE_224x224_greyscale_6secWindow_231023

README.md

@@ -1,5 +1,7 @@
 ---
+base_model: /content/drive/MyDrive/Seizure_EEG_Research/ViT_Seizure_Detection
 tags:
+- image-classification
 - generated_from_trainer
 datasets:
 - arrow
@@ -12,7 +14,7 @@ model-index:
       name: Image Classification
       type: image-classification
     dataset:
-      name: arrow
+      name: beans
       type: arrow
       config: default
       split: test
@@ -20,7 +22,7 @@ model-index:
     metrics:
     - name: Matthews Correlation
       type: matthews_correlation
-      value: 0.4389188148304588
+      value: 0.41096197273922397
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -28,10 +30,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # ViT_Seizure_Detection
 
-This model was trained from scratch on the arrow dataset.
+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.
 It achieves the following results on the evaluation set:
-- Loss: 0.3495
-- Matthews Correlation: 0.4389
+- Loss: 0.1622
+- Matthews Correlation: 0.4110
 
 ## Model description
 

all_results.json

@@ -0,0 +1,13 @@
+{
+    "epoch": 4.0,
+    "eval_loss": 0.1622471660375595,
+    "eval_matthews_correlation": 0.41096197273922397,
+    "eval_runtime": 3886.261,
+    "eval_samples_per_second": 100.402,
+    "eval_steps_per_second": 12.55,
+    "total_flos": 2.524899799855159e+20,
+    "train_loss": 0.0028621005677307255,
+    "train_runtime": 12160.5208,
+    "train_samples_per_second": 267.939,
+    "train_steps_per_second": 4.187
+}

eval_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 4.0,
+    "eval_loss": 0.1622471660375595,
+    "eval_matthews_correlation": 0.41096197273922397,
+    "eval_runtime": 3886.261,
+    "eval_samples_per_second": 100.402,
+    "eval_steps_per_second": 12.55
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 4.0,
+    "total_flos": 2.524899799855159e+20,
+    "train_loss": 0.0028621005677307255,
+    "train_runtime": 12160.5208,
+    "train_samples_per_second": 267.939,
+    "train_steps_per_second": 4.187
+}