🍻 cheers

README.md

@@ -24,10 +24,10 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.827115559599636
+      value: 0.8362147406733395
     - name: F1
       type: f1
-      value: 0.7978723404255319
+      value: 0.803921568627451
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -37,12 +37,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the stool-image dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.3761
-- Auroc: 0.9189
-- Accuracy: 0.8271
-- Sensitivity: 0.8333
-- Specificty: 0.8228
-- F1: 0.7979
+- Loss: 0.3833
+- Auroc: 0.9133
+- Accuracy: 0.8362
+- Sensitivity: 0.82
+- Specificty: 0.8475
+- F1: 0.8039
 
 ## Model description
 
@@ -74,8 +74,8 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Auroc  | Accuracy | Sensitivity | Specificty | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:------:|:--------:|:-----------:|:----------:|:------:|
-| 0.3755        | 0.98  | 100  | 0.4455          | 0.8677 | 0.7757   | 0.6966      | 0.8458     | 0.7447 |
-| 0.309         | 1.96  | 200  | 0.4360          | 0.8754 | 0.8127   | 0.7528      | 0.8657     | 0.7906 |
+| 0.4174        | 0.98  | 100  | 0.4700          | 0.8665 | 0.7916   | 0.7753      | 0.8060     | 0.7775 |
+| 0.3544        | 1.96  | 200  | 0.4304          | 0.8706 | 0.8179   | 0.7247      | 0.9005     | 0.7890 |
 
 
 ### Framework versions

all_results.json

@@ -1,17 +1,17 @@
 {
     "epoch": 2.0,
-    "eval_accuracy": 0.9320388349514563,
-    "eval_auroc": 0.9441448382126347,
-    "eval_eval_specificty": 0.8863636363636364,
-    "eval_f1": 0.9176470588235294,
-    "eval_loss": 0.37609395384788513,
-    "eval_runtime": 70.0505,
-    "eval_samples_per_second": 15.689,
-    "eval_sensitivity": 0.8863636363636364,
-    "eval_specificty": 0.8228043143297381,
-    "eval_steps_per_second": 1.97,
-    "train_loss": 0.4438221442933176,
-    "train_runtime": 152.958,
-    "train_samples_per_second": 21.248,
-    "train_steps_per_second": 1.334
+    "eval_accuracy": 0.9223300970873787,
+    "eval_auroc": 0.9345146379044684,
+    "eval_eval_specificty": 0.9090909090909091,
+    "eval_f1": 0.9090909090909091,
+    "eval_loss": 0.3832913041114807,
+    "eval_runtime": 68.0836,
+    "eval_samples_per_second": 16.142,
+    "eval_sensitivity": 0.9090909090909091,
+    "eval_specificty": 0.847457627118644,
+    "eval_steps_per_second": 2.027,
+    "train_loss": 0.4515615520524044,
+    "train_runtime": 168.2375,
+    "train_samples_per_second": 19.318,
+    "train_steps_per_second": 1.213
 }

eval_results.json

@@ -1,13 +1,13 @@
 {
     "epoch": 2.0,
-    "eval_accuracy": 0.9320388349514563,
-    "eval_auroc": 0.9441448382126347,
-    "eval_eval_specificty": 0.8863636363636364,
-    "eval_f1": 0.9176470588235294,
-    "eval_loss": 0.37609395384788513,
-    "eval_runtime": 70.0505,
-    "eval_samples_per_second": 15.689,
-    "eval_sensitivity": 0.8863636363636364,
-    "eval_specificty": 0.8228043143297381,
-    "eval_steps_per_second": 1.97
+    "eval_accuracy": 0.9223300970873787,
+    "eval_auroc": 0.9345146379044684,
+    "eval_eval_specificty": 0.9090909090909091,
+    "eval_f1": 0.9090909090909091,
+    "eval_loss": 0.3832913041114807,
+    "eval_runtime": 68.0836,
+    "eval_samples_per_second": 16.142,
+    "eval_sensitivity": 0.9090909090909091,
+    "eval_specificty": 0.847457627118644,
+    "eval_steps_per_second": 2.027
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:49e65a419570bcc528d2259195bf6cef77c24c492a4522dbe45e38c726a6a8a1
+oid sha256:542a2f033a1f2649f7e047d4c59568e96eed08c7708a219b5787d7a601d813fd
 size 343223968

runs/Jan06_12-44-27_DESKTOP-BDBS5RV/events.out.tfevents.1704512668.DESKTOP-BDBS5RV

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a66d51f146715d8189d2153b6885049a003a796a95370195b61aa937eb4e7340
+size 8852

runs/Jan06_12-44-27_DESKTOP-BDBS5RV/events.out.tfevents.1704512904.DESKTOP-BDBS5RV

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d05cd0d7b2a9b99bc29b7ba05085461b3db05045b32df305115ac1fcc5be815d
+size 40

train_results.json

@@ -1,7 +1,7 @@
 {
     "epoch": 2.0,
-    "train_loss": 0.4438221442933176,
-    "train_runtime": 152.958,
-    "train_samples_per_second": 21.248,
-    "train_steps_per_second": 1.334
+    "train_loss": 0.4515615520524044,
+    "train_runtime": 168.2375,
+    "train_samples_per_second": 19.318,
+    "train_steps_per_second": 1.213
 }

trainer_state.json

@@ -1,5 +1,5 @@
 {
-  "best_metric": 0.8754262395885739,
+  "best_metric": 0.8706327911006765,
   "best_model_checkpoint": "./stool-condition-classification\\checkpoint-200",
   "epoch": 2.0,
   "eval_steps": 100,
@@ -11,157 +11,157 @@
     {
       "epoch": 0.1,
       "learning_rate": 0.00019019607843137254,
-      "loss": 0.6003,
+      "loss": 0.593,
       "step": 10
     },
     {
       "epoch": 0.2,
       "learning_rate": 0.0001803921568627451,
-      "loss": 0.5057,
+      "loss": 0.4643,
       "step": 20
     },
     {
       "epoch": 0.29,
       "learning_rate": 0.00017058823529411766,
-      "loss": 0.492,
+      "loss": 0.5134,
       "step": 30
     },
     {
       "epoch": 0.39,
       "learning_rate": 0.00016078431372549022,
-      "loss": 0.5363,
+      "loss": 0.5355,
       "step": 40
     },
     {
       "epoch": 0.49,
       "learning_rate": 0.00015098039215686275,
-      "loss": 0.5434,
+      "loss": 0.4345,
       "step": 50
     },
     {
       "epoch": 0.59,
       "learning_rate": 0.0001411764705882353,
-      "loss": 0.4924,
+      "loss": 0.5371,
       "step": 60
     },
     {
       "epoch": 0.69,
       "learning_rate": 0.00013137254901960784,
-      "loss": 0.4675,
+      "loss": 0.4913,
       "step": 70
     },
     {
       "epoch": 0.78,
       "learning_rate": 0.00012156862745098039,
-      "loss": 0.4547,
+      "loss": 0.4804,
       "step": 80
     },
     {
       "epoch": 0.88,
       "learning_rate": 0.00011176470588235294,
-      "loss": 0.4565,
+      "loss": 0.4073,
       "step": 90
     },
     {
       "epoch": 0.98,
       "learning_rate": 0.00010196078431372549,
-      "loss": 0.3755,
+      "loss": 0.4174,
       "step": 100
     },
     {
       "epoch": 0.98,
-      "eval_accuracy": 0.7757255936675461,
-      "eval_auroc": 0.8676561015148975,
-      "eval_f1": 0.7447447447447447,
-      "eval_loss": 0.4455398917198181,
-      "eval_runtime": 19.6185,
-      "eval_samples_per_second": 19.318,
-      "eval_sensitivity": 0.6966292134831461,
-      "eval_specificty": 0.845771144278607,
-      "eval_steps_per_second": 2.447,
+      "eval_accuracy": 0.7915567282321899,
+      "eval_auroc": 0.8665241209681928,
+      "eval_f1": 0.7774647887323943,
+      "eval_loss": 0.4700045883655548,
+      "eval_runtime": 21.7128,
+      "eval_samples_per_second": 17.455,
+      "eval_sensitivity": 0.7752808988764045,
+      "eval_specificty": 0.8059701492537313,
+      "eval_steps_per_second": 2.211,
       "step": 100
     },
     {
       "epoch": 1.08,
       "learning_rate": 9.215686274509804e-05,
-      "loss": 0.4789,
+      "loss": 0.4938,
       "step": 110
     },
     {
       "epoch": 1.18,
       "learning_rate": 8.23529411764706e-05,
-      "loss": 0.3842,
+      "loss": 0.3797,
       "step": 120
     },
     {
       "epoch": 1.27,
       "learning_rate": 7.254901960784314e-05,
-      "loss": 0.5317,
+      "loss": 0.4885,
       "step": 130
     },
     {
       "epoch": 1.37,
       "learning_rate": 6.274509803921569e-05,
-      "loss": 0.4017,
+      "loss": 0.374,
       "step": 140
     },
     {
       "epoch": 1.47,
       "learning_rate": 5.294117647058824e-05,
-      "loss": 0.3627,
+      "loss": 0.487,
       "step": 150
     },
     {
       "epoch": 1.57,
       "learning_rate": 4.313725490196079e-05,
-      "loss": 0.4424,
+      "loss": 0.4517,
       "step": 160
     },
     {
       "epoch": 1.67,
       "learning_rate": 3.3333333333333335e-05,
-      "loss": 0.3926,
+      "loss": 0.4386,
       "step": 170
     },
     {
       "epoch": 1.76,
       "learning_rate": 2.3529411764705884e-05,
-      "loss": 0.4004,
+      "loss": 0.3918,
       "step": 180
     },
     {
       "epoch": 1.86,
       "learning_rate": 1.3725490196078432e-05,
-      "loss": 0.3161,
+      "loss": 0.3675,
       "step": 190
     },
     {
       "epoch": 1.96,
       "learning_rate": 3.92156862745098e-06,
-      "loss": 0.309,
+      "loss": 0.3544,
       "step": 200
     },
     {
       "epoch": 1.96,
-      "eval_accuracy": 0.8126649076517151,
-      "eval_auroc": 0.8754262395885739,
-      "eval_f1": 0.7905604719764012,
-      "eval_loss": 0.4360436201095581,
-      "eval_runtime": 19.2175,
-      "eval_samples_per_second": 19.722,
-      "eval_sensitivity": 0.7528089887640449,
-      "eval_specificty": 0.8656716417910447,
-      "eval_steps_per_second": 2.498,
+      "eval_accuracy": 0.8179419525065963,
+      "eval_auroc": 0.8706327911006765,
+      "eval_f1": 0.7889908256880735,
+      "eval_loss": 0.43039947748184204,
+      "eval_runtime": 19.7359,
+      "eval_samples_per_second": 19.204,
+      "eval_sensitivity": 0.7247191011235955,
+      "eval_specificty": 0.900497512437811,
+      "eval_steps_per_second": 2.432,
       "step": 200
     },
     {
       "epoch": 2.0,
       "step": 204,
       "total_flos": 2.51848966247424e+17,
-      "train_loss": 0.4438221442933176,
-      "train_runtime": 152.958,
-      "train_samples_per_second": 21.248,
-      "train_steps_per_second": 1.334
+      "train_loss": 0.4515615520524044,
+      "train_runtime": 168.2375,
+      "train_samples_per_second": 19.318,
+      "train_steps_per_second": 1.213
     }
   ],
   "logging_steps": 10,

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1fca6169b13c98d2024034ca2a018b9c8536fee89f8afeb3b4867f7e487a0d5b
+oid sha256:9bc96d212a9ea52203813b093d084ed31c1eb39d57b8b6759dbda959651c9550
 size 4283