Upload folder using huggingface_hub

best_model.pth

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

class_to_idx.json

@@ -0,0 +1,4 @@
+{
+  "clinical_macroscopic": 0,
+  "dermoscopic": 1
+}

training_config.json

@@ -0,0 +1,27 @@
+{
+  "dataset_index": "outputs/model/image_type_classifier_dataset_index.csv",
+  "output_dir": "models/image_type_classifier_v1",
+  "epochs": 10,
+  "batch_size": 32,
+  "lr": 0.0001,
+  "weight_decay": 0.01,
+  "backbone": "efficientnet_b0",
+  "image_size": 224,
+  "seed": 42,
+  "num_workers": 0,
+  "augmentation": "mild_clinical",
+  "max_train_batches": 60,
+  "max_val_batches": 60,
+  "class_names": [
+    "clinical_macroscopic",
+    "dermoscopic"
+  ],
+  "class_to_idx": {
+    "clinical_macroscopic": 0,
+    "dermoscopic": 1
+  },
+  "device": "mps",
+  "trainable_params": 4010110,
+  "train_examples": 19324,
+  "val_examples": 4153
+}

training_history.csv

@@ -0,0 +1,11 @@
+epoch,learning_rate,train_loss,train_accuracy,val_loss,val_accuracy,val_macro_f1,val_balanced_accuracy,val_recall_clinical_macroscopic,val_recall_dermoscopic
+1,0.0001,0.244332,0.930208,0.047333,0.992708,0.992173,0.993049,0.994334,0.991763
+2,0.0001,0.060335,0.984375,0.022223,0.994792,0.994386,0.993214,0.987252,0.999176
+3,0.0001,0.035836,0.990104,0.014861,0.996354,0.996074,0.995339,0.991501,0.999176
+4,0.0001,0.025463,0.992708,0.011616,0.997917,0.997759,0.997463,0.995751,0.999176
+5,0.0001,0.030131,0.991146,0.009411,0.998437,0.998319,0.998172,0.997167,0.999176
+6,0.0001,0.020469,0.992708,0.011868,0.997396,0.997196,0.996459,0.992918,1.0
+7,0.0001,0.016264,0.996354,0.009673,0.997917,0.997757,0.997167,0.994334,1.0
+8,0.0001,0.015343,0.996354,0.004977,0.998958,0.998881,0.999176,1.0,0.998353
+9,0.0001,0.016676,0.99375,0.005874,0.998958,0.998879,0.998584,0.997167,1.0
+10,0.0001,0.007457,0.998958,0.004117,0.999479,0.99944,0.999292,0.998584,1.0

training_metrics.json

@@ -0,0 +1,9 @@
+{
+  "best_epoch": 10,
+  "best_val_macro_f1": 0.9994397953619141,
+  "best_val_accuracy": 0.999479,
+  "best_val_balanced_accuracy": 0.999292,
+  "val_recall_clinical_macroscopic": 0.998584,
+  "val_recall_dermoscopic": 1.0,
+  "total_epochs": 10
+}