Upload TFBertForSequenceClassification

README.md

@@ -4,22 +4,18 @@ base_model: bert-base-uncased
 tags:
 - generated_from_keras_callback
 model-index:
-- name: zanetworker/quick-model
+- name: quick-model
   results: []
 ---
 
 <!-- This model card has been generated automatically according to the information Keras had access to. You should
 probably proofread and complete it, then remove this comment. -->
 
-# zanetworker/quick-model
+# quick-model
 
 This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 0.2655
-- Train Accuracy: 0.8948
-- Validation Loss: 0.3810
-- Validation Accuracy: 0.8480
-- Epoch: 1
+
 
 ## Model description
 
@@ -38,15 +34,11 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- optimizer: {'name': 'Adam', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 5e-05, 'decay_steps': 1377, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
+- optimizer: None
 - training_precision: float32
 
 ### Training results
 
-| Train Loss | Train Accuracy | Validation Loss | Validation Accuracy | Epoch |
-|:----------:|:--------------:|:---------------:|:-------------------:|:-----:|
-| 0.5277     | 0.7380         | 0.3773          | 0.8431              | 0     |
-| 0.2655     | 0.8948         | 0.3810          | 0.8480              | 1     |
 
 
 ### Framework versions

config.json

@@ -9,8 +9,16 @@
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
+  "id2label": {
+    "0": "not_equivalent",
+    "1": "equivalent"
+  },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
+  "label2id": {
+    "equivalent": 1,
+    "not_equivalent": 0
+  },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
   "model_type": "bert",

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7674677b7159fde804d2a9ae4add163160f7c897d59891c241c6d7b6078ca9fd
+oid sha256:6d7db882fa0deff51549b4cac48b4e1ef4af3d123b7d85db03c281b83c4a53b0
 size 438223128