Add evaluation results and output files to model card

README.md

@@ -5,18 +5,18 @@ tags:
   - model
   - evaluation
 metrics:
-  - average_precision: 0.93
-  - roc_auc: 0.90
-  - best threshold according to F1: 0.37
+  - average_precision: 0.83
+  - roc_auc: 0.79
+  - best threshold according to F1: 0.47
 ---
 
 # Binary Classification Model
 
 ## Evaluation Results
 
-**Average Precision:** 0.93  
-**ROC AUC:** 0.90
-**best threshold according to F1: 0.37
+**Average Precision:** 0.83  
+**ROC AUC:** 0.79
+**best threshold according to F1: 0.47
 
 
 ## Visualizations

checkpoint-8/config.json

@@ -0,0 +1,23 @@
+{
+  "_name_or_path": "GenetikaPlus/binary_classification_model_v3.1.5_Junction_batch2",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 256,
+  "image_size": 64,
+  "initializer_range": 0.02,
+  "intermediate_size": 512,
+  "layer_norm_eps": 1e-12,
+  "model_type": "vit",
+  "num_attention_heads": 8,
+  "num_channels": 3,
+  "num_hidden_layers": 4,
+  "patch_size": 4,
+  "qkv_bias": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.48.0"
+}

checkpoint-8/model.safetensors

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

checkpoint-8/optimizer.pt

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

checkpoint-8/preprocessor_config.json

@@ -0,0 +1,23 @@
+{
+  "do_convert_rgb": null,
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": false,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "ViTFeatureExtractor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 32,
+    "width": 32
+  }
+}

checkpoint-8/rng_state.pth

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

checkpoint-8/scheduler.pt

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

checkpoint-8/trainer_state.json

@@ -0,0 +1,113 @@
+{
+  "best_metric": 0.7894614129167639,
+  "best_model_checkpoint": "/tmp/logs/binary_classification_model_v3.1.5_Junction_batch2/checkpoint-8",
+  "epoch": 1.0,
+  "eval_steps": 500,
+  "global_step": 8,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.13333333333333333,
+      "grad_norm": 28787.228515625,
+      "learning_rate": 0.00035714285714285714,
+      "loss": 0.3856,
+      "step": 1
+    },
+    {
+      "epoch": 0.26666666666666666,
+      "grad_norm": 27319.6640625,
+      "learning_rate": 0.0007142857142857143,
+      "loss": 0.3913,
+      "step": 2
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 764010.0625,
+      "learning_rate": 0.0010714285714285715,
+      "loss": 0.657,
+      "step": 3
+    },
+    {
+      "epoch": 0.5333333333333333,
+      "grad_norm": 460677.375,
+      "learning_rate": 0.0014285714285714286,
+      "loss": 0.6085,
+      "step": 4
+    },
+    {
+      "epoch": 0.6666666666666666,
+      "grad_norm": 349255.0625,
+      "learning_rate": 0.0017857142857142859,
+      "loss": 0.6177,
+      "step": 5
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 162546.3125,
+      "learning_rate": 0.002142857142857143,
+      "loss": 0.5485,
+      "step": 6
+    },
+    {
+      "epoch": 0.9333333333333333,
+      "grad_norm": 441874.4375,
+      "learning_rate": 0.0025,
+      "loss": 0.7531,
+      "step": 7
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 153196.75,
+      "learning_rate": 0.002857142857142857,
+      "loss": 0.3116,
+      "step": 8
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.7538838920686836,
+      "eval_best_threshold": 0.5175232887268066,
+      "eval_f1": 0.7894614129167639,
+      "eval_loss": 0.5186505913734436,
+      "eval_pr_auc": 0.8433328368486523,
+      "eval_precision": 0.7086647132691503,
+      "eval_recall": 0.8910526315789473,
+      "eval_roc_auc": 0.8432376305376215,
+      "eval_runtime": 2.5983,
+      "eval_samples_per_second": 1412.087,
+      "eval_steps_per_second": 3.079,
+      "step": 8
+    }
+  ],
+  "logging_steps": 1,
+  "max_steps": 1400,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 200,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "EarlyStoppingCallback": {
+      "args": {
+        "early_stopping_patience": 20,
+        "early_stopping_threshold": 0.0
+      },
+      "attributes": {
+        "early_stopping_patience_counter": 0
+      }
+    },
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 2366877807673344.0,
+  "train_batch_size": 1024,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-8/training_args.bin

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

config.json

@@ -1,4 +1,5 @@
 {
+  "_name_or_path": "GenetikaPlus/binary_classification_model_v3.1.5_Junction_batch2",
   "architectures": [
     "ViTForImageClassification"
   ],

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:19a6f54899e78fbef1b7ba0c10d4ae45fb33aeb499d950a1c219df118f36cbf8
+oid sha256:dcf15dfe1603919b6569b7a8b9543d2d04a2c6071913433b787ac4a06829caba
 size 8760128

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d23bd679f2ac4710cc47143fd0d3ede7ba63efa5f001551630286ba32462adec
-size 5432
+oid sha256:99c19006956dac46c4f9af44fd85d92055df85ce6d52f69565d2637981ffd87f
+size 5496