Upload TFDistilBertForSequenceClassification

README.md

@@ -0,0 +1,49 @@
+---
+license: apache-2.0
+base_model: distilbert-base-uncased
+tags:
+- generated_from_keras_callback
+model-index:
+- name: distilbert-ft-ttvsplit
+  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. -->
+
+# distilbert-ft-ttvsplit
+
+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:
+
+
+## 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:
+- optimizer: {'name': 'Adam', 'weight_decay': None, 'clipnorm': None, 'global_clipnorm': None, 'clipvalue': None, 'use_ema': False, 'ema_momentum': 0.99, 'ema_overwrite_frequency': None, 'jit_compile': True, 'is_legacy_optimizer': False, 'learning_rate': 1e-05, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False}
+- training_precision: float32
+
+### Training results
+
+
+
+### Framework versions
+
+- Transformers 4.38.2
+- TensorFlow 2.15.0
+- Datasets 2.18.0
+- Tokenizers 0.15.2

config.json

@@ -0,0 +1,127 @@
+{
+  "_name_or_path": "distilbert-base-uncased",
+  "activation": "gelu",
+  "architectures": [
+    "DistilBertForSequenceClassification"
+  ],
+  "attention_dropout": 0.1,
+  "dim": 768,
+  "dropout": 0.1,
+  "hidden_dim": 3072,
+  "id2label": {
+    "0": "ALUMINIUM CAN",
+    "1": "ALUMINIUM SHEET",
+    "2": "ALUMINIUM TRAY",
+    "3": "CIGARETTE BUTT",
+    "4": "CIGARETTE PACK",
+    "5": "COMPOSTABLE PACKAGING",
+    "6": "CONDIMENT PACKETS",
+    "7": "COVID TEST",
+    "8": "CRUMBLED TISSUE",
+    "9": "CYLINDRICAL BATTERY",
+    "10": "FACE MASK",
+    "11": "GLASS BOTTLE",
+    "12": "GLASS JAR",
+    "13": "LAPTOP CHARGER",
+    "14": "MEDS BLISTER",
+    "15": "METAL CAP",
+    "16": "MIXED PAPER-PLASTIC PACKAGING",
+    "17": "ORGANIC SCRAPS",
+    "18": "PAPER BOWL",
+    "19": "PAPER CUP",
+    "20": "PAPER FOOD PACKAGING",
+    "21": "PAPER MAGASINE",
+    "22": "PAPER PACKAGING",
+    "23": "PAPER PLATE",
+    "24": "PAPER SHEET",
+    "25": "PAPER SUGAR BAG",
+    "26": "PAPER TRAY",
+    "27": "PHONE CHARGER",
+    "28": "PIZZA BOX",
+    "29": "PLASTIC BAG",
+    "30": "PLASTIC BOTTLE",
+    "31": "PLASTIC BOWL",
+    "32": "PLASTIC CAP",
+    "33": "PLASTIC CUP",
+    "34": "PLASTIC CUTLERY",
+    "35": "PLASTIC DISH",
+    "36": "PLASTIC GLOVES",
+    "37": "PLASTIC PACKAGING",
+    "38": "PLASTIC SNACK PACKAGING",
+    "39": "PLASTIC STICKS",
+    "40": "PLASTIC STRAW",
+    "41": "PLASTIC TRAY",
+    "42": "RECEIPT",
+    "43": "SMARTPHONE",
+    "44": "TEA BAG",
+    "45": "TETRAPACK",
+    "46": "TOBACCO PACK",
+    "47": "TRANSPORT TICKET",
+    "48": "WOODEN CUTLERY",
+    "49": "WOODEN STICKS"
+  },
+  "initializer_range": 0.02,
+  "label2id": {
+    "ALUMINIUM CAN": 0,
+    "ALUMINIUM SHEET": 1,
+    "ALUMINIUM TRAY": 2,
+    "CIGARETTE BUTT": 3,
+    "CIGARETTE PACK": 4,
+    "COMPOSTABLE PACKAGING": 5,
+    "CONDIMENT PACKETS": 6,
+    "COVID TEST": 7,
+    "CRUMBLED TISSUE": 8,
+    "CYLINDRICAL BATTERY": 9,
+    "FACE MASK": 10,
+    "GLASS BOTTLE": 11,
+    "GLASS JAR": 12,
+    "LAPTOP CHARGER": 13,
+    "MEDS BLISTER": 14,
+    "METAL CAP": 15,
+    "MIXED PAPER-PLASTIC PACKAGING": 16,
+    "ORGANIC SCRAPS": 17,
+    "PAPER BOWL": 18,
+    "PAPER CUP": 19,
+    "PAPER FOOD PACKAGING": 20,
+    "PAPER MAGASINE": 21,
+    "PAPER PACKAGING": 22,
+    "PAPER PLATE": 23,
+    "PAPER SHEET": 24,
+    "PAPER SUGAR BAG": 25,
+    "PAPER TRAY": 26,
+    "PHONE CHARGER": 27,
+    "PIZZA BOX": 28,
+    "PLASTIC BAG": 29,
+    "PLASTIC BOTTLE": 30,
+    "PLASTIC BOWL": 31,
+    "PLASTIC CAP": 32,
+    "PLASTIC CUP": 33,
+    "PLASTIC CUTLERY": 34,
+    "PLASTIC DISH": 35,
+    "PLASTIC GLOVES": 36,
+    "PLASTIC PACKAGING": 37,
+    "PLASTIC SNACK PACKAGING": 38,
+    "PLASTIC STICKS": 39,
+    "PLASTIC STRAW": 40,
+    "PLASTIC TRAY": 41,
+    "RECEIPT": 42,
+    "SMARTPHONE": 43,
+    "TEA BAG": 44,
+    "TETRAPACK": 45,
+    "TOBACCO PACK": 46,
+    "TRANSPORT TICKET": 47,
+    "WOODEN CUTLERY": 48,
+    "WOODEN STICKS": 49
+  },
+  "max_position_embeddings": 512,
+  "model_type": "distilbert",
+  "n_heads": 12,
+  "n_layers": 6,
+  "pad_token_id": 0,
+  "qa_dropout": 0.1,
+  "seq_classif_dropout": 0.2,
+  "sinusoidal_pos_embds": false,
+  "tie_weights_": true,
+  "transformers_version": "4.38.2",
+  "vocab_size": 30522
+}

tf_model.h5

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