End of training

README.md

@@ -32,8 +32,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the imdb dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.7584
 - Accuracy: 0.9273
+- Loss: 0.7584
 
 ## Model description
 
@@ -66,14 +66,14 @@ The following hyperparameters were used during training:
 |:-------------:|:-----:|:-----:|:--------:|:---------------:|
 | 0.0209        | 1.0   | 1563  | 0.9263   | 0.5386          |
 | 0.0           | 2.0   | 3126  | 0.9284   | 0.6249          |
-| 0.016         | 3.0   | 4689  | 0.6532   | 0.9166          |
-| 0.0125        | 4.0   | 6252  | 0.6454   | 0.9198          |
-| 0.0032        | 5.0   | 7815  | 0.6582   | 0.9264          |
-| 0.0139        | 6.0   | 9378  | 0.6766   | 0.9260          |
-| 0.0001        | 7.0   | 10941 | 0.7319   | 0.9283          |
-| 0.0           | 8.0   | 12504 | 0.7658   | 0.9269          |
-| 0.0132        | 9.0   | 14067 | 0.7416   | 0.9266          |
-| 0.0           | 10.0  | 15630 | 0.7584   | 0.9273          |
+| 0.016         | 3.0   | 4689  | 0.9166   | 0.6532          |
+| 0.0125        | 4.0   | 6252  | 0.9198   | 0.6454          |
+| 0.0032        | 5.0   | 7815  | 0.9264   | 0.6582          |
+| 0.0139        | 6.0   | 9378  | 0.9260   | 0.6766          |
+| 0.0001        | 7.0   | 10941 | 0.9283   | 0.7319          |
+| 0.0           | 8.0   | 12504 | 0.9269   | 0.7658          |
+| 0.0132        | 9.0   | 14067 | 0.9266   | 0.7416          |
+| 0.0           | 10.0  | 15630 | 0.9273   | 0.7584          |
 
 
 ### Framework versions

config.json

@@ -22,7 +22,6 @@
   "n_heads": 12,
   "n_layers": 6,
   "pad_token_id": 0,
-  "problem_type": "single_label_classification",
   "qa_dropout": 0.1,
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,

runs/Nov20_08-21-08_HAN/events.out.tfevents.1700439678.HAN.8728.10

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:591901b51a749c1ef4aff72b6378370178efa45ccd2ca48868683398c7445c6c
-size 4091
+oid sha256:a9528c02646de77e2b3aa5823bb3c82d0bc8b87be3ea67b702767f5c79bd7c14
+size 4155