End of training

README.md

@@ -0,0 +1,62 @@
+---
+license: apache-2.0
+base_model: distilbert-base-uncased
+tags:
+- generated_from_trainer
+metrics:
+- accuracy
+model-index:
+- name: distilbert-imdb
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# distilbert-imdb
+
+This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.1248
+- Accuracy: 0.97
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 5e-05
+- train_batch_size: 16
+- eval_batch_size: 16
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- num_epochs: 3
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 0.0812        | 1.0   | 313  | 0.1134          | 0.966    |
+| 0.0386        | 2.0   | 626  | 0.1193          | 0.9705   |
+| 0.0007        | 3.0   | 939  | 0.1248          | 0.97     |
+
+
+### Framework versions
+
+- Transformers 4.36.1
+- Pytorch 2.1.0+cu121
+- Datasets 2.15.0
+- Tokenizers 0.15.0

config.json

@@ -0,0 +1,25 @@
+{
+  "_name_or_path": "distilbert-base-uncased",
+  "activation": "gelu",
+  "architectures": [
+    "DistilBertForSequenceClassification"
+  ],
+  "attention_dropout": 0.1,
+  "dim": 768,
+  "dropout": 0.1,
+  "hidden_dim": 3072,
+  "initializer_range": 0.02,
+  "max_position_embeddings": 512,
+  "model_type": "distilbert",
+  "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,
+  "tie_weights_": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.36.1",
+  "vocab_size": 30522
+}

model.safetensors

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

runs/Dec18_06-09-16_9501bdc42b04/events.out.tfevents.1702879762.9501bdc42b04.4290.0

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

runs/Dec18_06-09-16_9501bdc42b04/events.out.tfevents.1702880077.9501bdc42b04.4290.1

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

runs/Dec18_06-16-35_9501bdc42b04/events.out.tfevents.1702880227.9501bdc42b04.4290.2

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

runs/Dec18_06-16-35_9501bdc42b04/events.out.tfevents.1702880422.9501bdc42b04.4290.3

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

runs/Dec18_06-16-35_9501bdc42b04/events.out.tfevents.1702880630.9501bdc42b04.4290.4

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

runs/Dec18_06-25-12_9501bdc42b04/events.out.tfevents.1702880723.9501bdc42b04.4290.5

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

runs/Dec18_06-25-12_9501bdc42b04/events.out.tfevents.1702880822.9501bdc42b04.4290.6

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

runs/Dec18_06-25-12_9501bdc42b04/events.out.tfevents.1702880924.9501bdc42b04.4290.7

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

runs/Dec18_06-30-08_9501bdc42b04/events.out.tfevents.1702881013.9501bdc42b04.4290.8

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

runs/Dec18_06-30-52_9501bdc42b04/events.out.tfevents.1702881056.9501bdc42b04.4290.9

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

runs/Dec18_06-56-22_9501bdc42b04/events.out.tfevents.1702882592.9501bdc42b04.16654.0

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

special_tokens_map.json

@@ -0,0 +1,7 @@
+{
+  "cls_token": "[CLS]",
+  "mask_token": "[MASK]",
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "unk_token": "[UNK]"
+}

tokenizer_config.json

@@ -0,0 +1,55 @@
+{
+  "added_tokens_decoder": {
+    "0": {
+      "content": "[PAD]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "100": {
+      "content": "[UNK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "101": {
+      "content": "[CLS]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "102": {
+      "content": "[SEP]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "103": {
+      "content": "[MASK]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "clean_up_tokenization_spaces": true,
+  "cls_token": "[CLS]",
+  "do_lower_case": true,
+  "mask_token": "[MASK]",
+  "model_max_length": 512,
+  "pad_token": "[PAD]",
+  "sep_token": "[SEP]",
+  "strip_accents": null,
+  "tokenize_chinese_chars": true,
+  "tokenizer_class": "DistilBertTokenizer",
+  "unk_token": "[UNK]"
+}

training_args.bin

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