End of training

README.md

@@ -15,8 +15,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1220
-- Accuracy: 0.9531
+- Loss: 0.1200
+- Accuracy: 0.9541
 
 ## Model description
 
@@ -50,11 +50,11 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.4165        | 0.99  | 52   | 0.2990          | 0.8902   |
-| 0.2961        | 2.0   | 105  | 0.2657          | 0.8974   |
-| 0.1728        | 2.99  | 157  | 0.1917          | 0.9177   |
-| 0.2614        | 4.0   | 210  | 0.2160          | 0.9165   |
-| 0.1238        | 4.95  | 260  | 0.1240          | 0.9499   |
+| 0.4084        | 0.99  | 52   | 0.3509          | 0.8628   |
+| 0.251         | 2.0   | 105  | 0.1811          | 0.9332   |
+| 0.1841        | 2.99  | 157  | 0.2061          | 0.9189   |
+| 0.198         | 4.0   | 210  | 0.1512          | 0.9391   |
+| 0.1229        | 4.95  | 260  | 0.1240          | 0.9547   |
 
 
 ### Framework versions

config.json

@@ -8,7 +8,6 @@
   },
   "input_channels": 3,
   "model_type": "alexnet",
-  "num_classes": 2,
   "torch_dtype": "float32",
   "transformers_version": "4.33.2"
 }

configuration_alexnet.py

@@ -3,13 +3,12 @@ from transformers import PretrainedConfig
 
 class AlexNetConfig(PretrainedConfig):
   model_type = "alexnet"
-  def __init__(self, **kwargs):    
-    self.num_classes = 2    
+  def __init__(self, id2label=None, label2id=None, labels=[], **kwargs):    
     self.input_channels = 3
     self.output_hidden_states = True
     self.return_dict = True
-    self.id2label={0: "normal", 1: "pneumonia"}
-    self.label2id={"normal": 0, "pneumonia": 1}
-    self.num_labels = 2
+    self.id2label=id2label
+    self.label2id=label2id
+    self.num_labels = len(labels)
     self.model_type = "alexnet"
     super().__init__(**kwargs)

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ad61d47816f85c94b70f61cc1123526099a9ff816cff43cbe9740a137f369f31
+oid sha256:81529019ecc95f51f2d7be8a08f9da16ef91ca7cc2261c29deb44934e1dac973
 size 233163169

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:65f1e54e241feacc3bb70e713d7bdb2ac39f27aa661eff0a7b3345d93d4fa843
+oid sha256:5328edd3fee2ca0c37eb4380a9a8b0df91e35c46aad555abd018bdf5336a3835
 size 4091