mstatt commited on
Commit
d10e252
1 Parent(s): 0202629

Upload DistilBertForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +17 -29
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "Falconsai/intent_classification",
3
  "activation": "gelu",
4
  "architectures": [
5
  "DistilBertForSequenceClassification"
@@ -9,36 +9,24 @@
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
  "id2label": {
12
- "0": "cancel order",
13
- "1": "change order",
14
- "2": "change shipping address",
15
- "3": "cancellation",
16
- "4": "check invoice",
17
- "5": "payment methods",
18
- "6": "refund policy",
19
- "7": "complaint",
20
- "8": "contact customer service",
21
- "9": "contact human agent",
22
- "10": "create account",
23
- "11": "delete account",
24
- "12": "delivery options",
25
- "13": "delivery period",
26
- "14": "edit account",
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:d90cc47103caf7cf8576e6cddb4a0e07aed962269d0b8447363e0fdda36d00bb
3
- size 267934061
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9827b446f7bbcaca6b56c2a8accecc97264953e9d5d9adaf7c29d6a6dad61f3e
3
+ size 267894125