thomasavare commited on
Commit
597155d
·
1 Parent(s): 13d71ae

Upload TFRobertaForSequenceClassification

Browse files
Files changed (3) hide show
  1. README.md +48 -0
  2. config.json +130 -0
  3. tf_model.h5 +3 -0
README.md ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - generated_from_keras_callback
5
+ model-index:
6
+ - name: distilroberta-ft-test1
7
+ results: []
8
+ ---
9
+
10
+ <!-- This model card has been generated automatically according to the information Keras had access to. You should
11
+ probably proofread and complete it, then remove this comment. -->
12
+
13
+ # distilroberta-ft-test1
14
+
15
+ This model is a fine-tuned version of [distilroberta-base](https://huggingface.co/distilroberta-base) on an unknown dataset.
16
+ It achieves the following results on the evaluation set:
17
+
18
+
19
+ ## Model description
20
+
21
+ More information needed
22
+
23
+ ## Intended uses & limitations
24
+
25
+ More information needed
26
+
27
+ ## Training and evaluation data
28
+
29
+ More information needed
30
+
31
+ ## Training procedure
32
+
33
+ ### Training hyperparameters
34
+
35
+ The following hyperparameters were used during training:
36
+ - 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': 5e-05, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False}
37
+ - training_precision: float32
38
+
39
+ ### Training results
40
+
41
+
42
+
43
+ ### Framework versions
44
+
45
+ - Transformers 4.28.1
46
+ - TensorFlow 2.12.0
47
+ - Datasets 2.11.0
48
+ - Tokenizers 0.13.3
config.json ADDED
@@ -0,0 +1,130 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "distilroberta-base",
3
+ "architectures": [
4
+ "RobertaForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "eos_token_id": 2,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.1,
12
+ "hidden_size": 768,
13
+ "id2label": {
14
+ "0": "ALUMINIUM CAN",
15
+ "1": "ALUMINIUM SHEET",
16
+ "2": "ALUMINIUM TRAY",
17
+ "3": "CIGARETTE BUTT",
18
+ "4": "CIGARETTE PACK",
19
+ "5": "COMPOSTABLE PACKAGING",
20
+ "6": "CONDIMENT PACKETS",
21
+ "7": "COVID TEST",
22
+ "8": "CRUMBLED TISSUE",
23
+ "9": "CYLINDRICAL BATTERY",
24
+ "10": "FACE MASK",
25
+ "11": "GLASS BOTTLE",
26
+ "12": "GLASS JAR",
27
+ "13": "LAPTOP CHARGER",
28
+ "14": "MEDS BLISTER",
29
+ "15": "METAL CAP",
30
+ "16": "MIXED PAPER-PLASTIC PACKAGING",
31
+ "17": "ORGANIC SCRAPS",
32
+ "18": "PAPER BOWL",
33
+ "19": "PAPER CUP",
34
+ "20": "PAPER FOOD PACKAGING",
35
+ "21": "PAPER MAGASINE",
36
+ "22": "PAPER PACKAGING",
37
+ "23": "PAPER PLATE",
38
+ "24": "PAPER SHEET",
39
+ "25": "PAPER SUGAR BAG",
40
+ "26": "PAPER TRAY",
41
+ "27": "PHONE CHARGER",
42
+ "28": "PIZZA BOX",
43
+ "29": "PLASTIC BAG",
44
+ "30": "PLASTIC BOTTLE",
45
+ "31": "PLASTIC BOWL",
46
+ "32": "PLASTIC CAP",
47
+ "33": "PLASTIC CUP",
48
+ "34": "PLASTIC CUTLERY",
49
+ "35": "PLASTIC DISH",
50
+ "36": "PLASTIC GLOVES",
51
+ "37": "PLASTIC PACKAGING",
52
+ "38": "PLASTIC SNACK PACKAGING",
53
+ "39": "PLASTIC STICKS",
54
+ "40": "PLASTIC STRAW",
55
+ "41": "PLASTIC TRAY",
56
+ "42": "RECEIPT",
57
+ "43": "SMARTPHONE",
58
+ "44": "TEA BAG",
59
+ "45": "TETRAPACK",
60
+ "46": "TOBACCO PACK",
61
+ "47": "TRANSPORT TICKET",
62
+ "48": "WOODEN CUTLERY",
63
+ "49": "WOODEN STICKS"
64
+ },
65
+ "initializer_range": 0.02,
66
+ "intermediate_size": 3072,
67
+ "label2id": {
68
+ "ALUMINIUM CAN": 0,
69
+ "ALUMINIUM SHEET": 1,
70
+ "ALUMINIUM TRAY": 2,
71
+ "CIGARETTE BUTT": 3,
72
+ "CIGARETTE PACK": 4,
73
+ "COMPOSTABLE PACKAGING": 5,
74
+ "CONDIMENT PACKETS": 6,
75
+ "COVID TEST": 7,
76
+ "CRUMBLED TISSUE": 8,
77
+ "CYLINDRICAL BATTERY": 9,
78
+ "FACE MASK": 10,
79
+ "GLASS BOTTLE": 11,
80
+ "GLASS JAR": 12,
81
+ "LAPTOP CHARGER": 13,
82
+ "MEDS BLISTER": 14,
83
+ "METAL CAP": 15,
84
+ "MIXED PAPER-PLASTIC PACKAGING": 16,
85
+ "ORGANIC SCRAPS": 17,
86
+ "PAPER BOWL": 18,
87
+ "PAPER CUP": 19,
88
+ "PAPER FOOD PACKAGING": 20,
89
+ "PAPER MAGASINE": 21,
90
+ "PAPER PACKAGING": 22,
91
+ "PAPER PLATE": 23,
92
+ "PAPER SHEET": 24,
93
+ "PAPER SUGAR BAG": 25,
94
+ "PAPER TRAY": 26,
95
+ "PHONE CHARGER": 27,
96
+ "PIZZA BOX": 28,
97
+ "PLASTIC BAG": 29,
98
+ "PLASTIC BOTTLE": 30,
99
+ "PLASTIC BOWL": 31,
100
+ "PLASTIC CAP": 32,
101
+ "PLASTIC CUP": 33,
102
+ "PLASTIC CUTLERY": 34,
103
+ "PLASTIC DISH": 35,
104
+ "PLASTIC GLOVES": 36,
105
+ "PLASTIC PACKAGING": 37,
106
+ "PLASTIC SNACK PACKAGING": 38,
107
+ "PLASTIC STICKS": 39,
108
+ "PLASTIC STRAW": 40,
109
+ "PLASTIC TRAY": 41,
110
+ "RECEIPT": 42,
111
+ "SMARTPHONE": 43,
112
+ "TEA BAG": 44,
113
+ "TETRAPACK": 45,
114
+ "TOBACCO PACK": 46,
115
+ "TRANSPORT TICKET": 47,
116
+ "WOODEN CUTLERY": 48,
117
+ "WOODEN STICKS": 49
118
+ },
119
+ "layer_norm_eps": 1e-05,
120
+ "max_position_embeddings": 514,
121
+ "model_type": "roberta",
122
+ "num_attention_heads": 12,
123
+ "num_hidden_layers": 6,
124
+ "pad_token_id": 1,
125
+ "position_embedding_type": "absolute",
126
+ "transformers_version": "4.28.1",
127
+ "type_vocab_size": 1,
128
+ "use_cache": true,
129
+ "vocab_size": 50265
130
+ }
tf_model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:452f21da46b13ad85b3d546562ac1e2d821271641c07f7ebdc096fa6212e3a4b
3
+ size 328776792