karoladelk/skin_cancer_classification

README.md

@@ -2,6 +2,7 @@
 license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
+- image-classification
 - generated_from_trainer
 metrics:
 - accuracy
@@ -15,10 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # skin_cancer_classification
 
-This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
+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 Pranavkpba2000/skin_cancer_dataset dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.9207
-- Accuracy: 0.8555
+- Loss: 0.5151
+- Accuracy: 0.8260
 
 ## Model description
 

all_results.json

@@ -0,0 +1,13 @@
+{
+    "epoch": 10.0,
+    "eval_accuracy": 0.8260380014074595,
+    "eval_loss": 0.5151475667953491,
+    "eval_runtime": 131.1738,
+    "eval_samples_per_second": 54.165,
+    "eval_steps_per_second": 3.392,
+    "total_flos": 2.209880408214012e+19,
+    "train_loss": 0.16937365282474154,
+    "train_runtime": 20769.0609,
+    "train_samples_per_second": 13.73,
+    "train_steps_per_second": 0.429
+}

eval_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 10.0,
+    "eval_accuracy": 0.8260380014074595,
+    "eval_loss": 0.5151475667953491,
+    "eval_runtime": 131.1738,
+    "eval_samples_per_second": 54.165,
+    "eval_steps_per_second": 3.392
+}

runs/Apr22_09-55-47_51b0a3d70cdb/events.out.tfevents.1713800932.51b0a3d70cdb.98.1

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

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 10.0,
+    "total_flos": 2.209880408214012e+19,
+    "train_loss": 0.16937365282474154,
+    "train_runtime": 20769.0609,
+    "train_samples_per_second": 13.73,
+    "train_steps_per_second": 0.429
+}