Upload TFDistilBertForSequenceClassification

README.md

@@ -1,7 +1,7 @@
 ---
-base_model: distilbert-base-uncased
 library_name: transformers
 license: apache-2.0
+base_model: distilbert-base-uncased
 tags:
 - generated_from_keras_callback
 model-index:
@@ -16,10 +16,10 @@ 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 an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 0.2656
-- Train Accuracy: 0.9728
-- Validation Loss: 0.3730
-- Validation Accuracy: 0.8500
+- Train Loss: 0.1733
+- Train Accuracy: 0.9756
+- Validation Loss: 0.3006
+- Validation Accuracy: 0.8977
 - Epoch: 6
 
 ## Model description
@@ -46,13 +46,13 @@ The following hyperparameters were used during training:
 
 | Train Loss | Train Accuracy | Validation Loss | Validation Accuracy | Epoch |
 |:----------:|:--------------:|:---------------:|:-------------------:|:-----:|
-| 0.6956     | 0.4728         | 0.6850          | 0.625               | 0     |
-| 0.6722     | 0.6685         | 0.6670          | 0.75                | 1     |
-| 0.6520     | 0.7120         | 0.6367          | 0.6875              | 2     |
-| 0.6002     | 0.8261         | 0.5741          | 0.8000              | 3     |
-| 0.5079     | 0.875          | 0.4797          | 0.8875              | 4     |
-| 0.3805     | 0.9511         | 0.4118          | 0.8250              | 5     |
-| 0.2656     | 0.9728         | 0.3730          | 0.8500              | 6     |
+| 0.6870     | 0.5707         | 0.6656          | 0.6136              | 0     |
+| 0.6542     | 0.6293         | 0.6289          | 0.6477              | 1     |
+| 0.5970     | 0.7902         | 0.5541          | 0.7955              | 2     |
+| 0.4936     | 0.8829         | 0.4490          | 0.8523              | 3     |
+| 0.3649     | 0.9415         | 0.3775          | 0.875               | 4     |
+| 0.2563     | 0.9561         | 0.3254          | 0.8977              | 5     |
+| 0.1733     | 0.9756         | 0.3006          | 0.8977              | 6     |
 
 
 ### Framework versions

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9341a2e684c8a723f5278c28604a0c759865d15ce9718dc7fcf272fad482f171
+oid sha256:a74d43b14980d59d36249fa00d6e2161b380d0b5c56ee5677def4d21fa682ba1
 size 267955144