Upload TFBertForTokenClassification

README.md

@@ -14,9 +14,7 @@ probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [bert-base-cased](https://huggingface.co/bert-base-cased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 0.0087
-- Validation Loss: 0.2588
-- Epoch: 9
+
 
 ## Model description
 
@@ -35,28 +33,16 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- optimizer: {'inner_optimizer': {'class_name': 'AdamWeightDecay', 'config': {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': 2890, '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-08, 'amsgrad': False, 'weight_decay_rate': 0.01}}, 'dynamic': True, 'initial_scale': 32768.0, 'dynamic_growth_steps': 2000}
+- optimizer: {'inner_optimizer': {'class_name': 'AdamWeightDecay', 'config': {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': 22700, '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-08, 'amsgrad': False, 'weight_decay_rate': 0.01}}, 'dynamic': True, 'initial_scale': 32768.0, 'dynamic_growth_steps': 2000}
 - training_precision: mixed_float16
 
 ### Training results
 
-| Train Loss | Validation Loss | Epoch |
-|:----------:|:---------------:|:-----:|
-| 0.5009     | 0.2286          | 0     |
-| 0.1843     | 0.1930          | 1     |
-| 0.1116     | 0.1836          | 2     |
-| 0.0686     | 0.1865          | 3     |
-| 0.0430     | 0.2524          | 4     |
-| 0.0311     | 0.2881          | 5     |
-| 0.0186     | 0.2460          | 6     |
-| 0.0156     | 0.2437          | 7     |
-| 0.0127     | 0.2700          | 8     |
-| 0.0087     | 0.2588          | 9     |
 
 
 ### Framework versions
 
-- Transformers 4.22.1
-- TensorFlow 2.8.2
-- Datasets 2.5.1
-- Tokenizers 0.12.1
+- Transformers 4.24.0
+- TensorFlow 2.9.2
+- Datasets 2.6.1
+- Tokenizers 0.13.2

config.json

@@ -13,6 +13,8 @@
     "0": "O",
     "1": "B-Brand",
     "10": "I-TM",
+    "11": "B-Color",
+    "12": "I-Color",
     "2": "I-Brand",
     "3": "B-Model",
     "4": "I-Model",
@@ -27,11 +29,13 @@
   "label2id": {
     "B-Brand": "1",
     "B-CC": "7",
+    "B-Color": "11",
     "B-Model": "3",
     "B-TM": "9",
     "B-Tipe": "5",
     "I-Brand": "2",
     "I-CC": "8",
+    "I-Color": "12",
     "I-Model": "4",
     "I-TM": "10",
     "I-Tipe": "6",
@@ -44,7 +48,7 @@
   "num_hidden_layers": 12,
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
-  "transformers_version": "4.22.1",
+  "transformers_version": "4.24.0",
   "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 28996

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d7dd0afd67966a8f4adf4668b57b2e0e8307a7843264002c1965f82b94115b55
-size 431198260
+oid sha256:b1a1939fe81972589c6743fbff06d57d2427a59b478b4b22d618c6983a5690a1
+size 431204412