Upload DistilBertForSequenceClassification
Browse files- config.json +17 -29
- pytorch_model.bin +2 -2
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"activation": "gelu",
|
4 |
"architectures": [
|
5 |
"DistilBertForSequenceClassification"
|
@@ -9,36 +9,24 @@
|
|
9 |
"dropout": 0.1,
|
10 |
"hidden_dim": 3072,
|
11 |
"id2label": {
|
12 |
-
"0": "
|
13 |
-
"1": "
|
14 |
-
"2": "
|
15 |
-
"3": "
|
16 |
-
"4": "
|
17 |
-
"5": "
|
18 |
-
"6": "
|
19 |
-
"7": "
|
20 |
-
"8": "
|
21 |
-
"9": "
|
22 |
-
"10": "
|
23 |
-
"11": "
|
24 |
-
"12": "
|
25 |
-
"13": "
|
26 |
-
"14": "
|
27 |
-
"15": "get invoice",
|
28 |
-
"16": "get refund",
|
29 |
-
"17": "subscription",
|
30 |
-
"18": "payment issue",
|
31 |
-
"19": "place order",
|
32 |
-
"20": "recover password",
|
33 |
-
"21": "registration problems",
|
34 |
-
"22": "review",
|
35 |
-
"23": "set up shipping address",
|
36 |
-
"24": "switch account",
|
37 |
-
"25": "track order",
|
38 |
-
"26": "track refund",
|
39 |
-
"27": "appointment"
|
40 |
},
|
41 |
"initializer_range": 0.02,
|
|
|
42 |
"max_position_embeddings": 512,
|
43 |
"model_type": "distilbert",
|
44 |
"n_heads": 12,
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "./model/",
|
3 |
"activation": "gelu",
|
4 |
"architectures": [
|
5 |
"DistilBertForSequenceClassification"
|
|
|
9 |
"dropout": 0.1,
|
10 |
"hidden_dim": 3072,
|
11 |
"id2label": {
|
12 |
+
"0": "cancellation",
|
13 |
+
"1": "ordering",
|
14 |
+
"2": "shipping",
|
15 |
+
"3": "invoicing",
|
16 |
+
"4": "billing and payment",
|
17 |
+
"5": "returns and refunds",
|
18 |
+
"6": "complaints and feedback",
|
19 |
+
"7": "speak to person",
|
20 |
+
"8": "edit account",
|
21 |
+
"9": "delete account",
|
22 |
+
"10": "delivery information",
|
23 |
+
"11": "subscription",
|
24 |
+
"12": "recover password",
|
25 |
+
"13": "registration problems",
|
26 |
+
"14": "appointment"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
27 |
},
|
28 |
"initializer_range": 0.02,
|
29 |
+
"label2id": null,
|
30 |
"max_position_embeddings": 512,
|
31 |
"model_type": "distilbert",
|
32 |
"n_heads": 12,
|
pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9827b446f7bbcaca6b56c2a8accecc97264953e9d5d9adaf7c29d6a6dad61f3e
|
3 |
+
size 267894125
|