End of training

README.md

@@ -5,9 +5,28 @@ tags:
 - generated_from_trainer
 datasets:
 - imdb
+metrics:
+- accuracy
+- f1
 model-index:
 - name: pruned_model
-  results: []
+  results:
+  - task:
+      name: Text Classification
+      type: text-classification
+    dataset:
+      name: imdb
+      type: imdb
+      config: plain_text
+      split: test
+      args: plain_text
+    metrics:
+    - name: Accuracy
+      type: accuracy
+      value: 0.83
+    - name: F1
+      type: f1
+      value: 0.8338762214983714
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -16,6 +35,10 @@ should probably proofread and complete it, then remove this comment. -->
 # pruned_model
 
 This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the imdb dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.4037
+- Accuracy: 0.83
+- F1: 0.8339
 
 ## Model description
 

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:471ab3359ce68584cb824c2e27d4b13ec930e0feff6fd9ce165bfc7639bfdc53
-size 437964888
+oid sha256:9b45098d86f942dade1830028ddc8c6037268b2f0e39d4874147492057da83a3
+size 267838384

runs/Nov23_02-14-28_a4a173ca1381/events.out.tfevents.1700705669.a4a173ca1381.4271.0

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

runs/Nov23_02-14-28_a4a173ca1381/events.out.tfevents.1700710963.a4a173ca1381.4271.1

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7ca1a83474802d0ba8af01e822df3d5b96f42a6f2282f2412ad37fd47e72fcf3
+oid sha256:5ac8e50b96c1f0efe41bd9320de25b038f5bc0cbfe4503f8fd10dce503a2c40b
 size 4664