Updated the tokenizer parameters and made some changes to the labels.

Previous version would only predict the labels as numbers. This version will show the corresponding category of the predicted label.

config.json

@@ -9,37 +9,37 @@
  "dropout": 0.1,
  "hidden_dim": 3072,
  "id2label": {
- "0": "LABEL_0",
- "1": "LABEL_1",
- "2": "LABEL_2",
- "3": "LABEL_3",
- "4": "LABEL_4",
- "5": "LABEL_5",
- "6": "LABEL_6",
- "7": "LABEL_7",
- "8": "LABEL_8",
- "9": "LABEL_9",
- "10": "LABEL_10",
- "11": "LABEL_11",
- "12": "LABEL_12",
- "13": "LABEL_13"
+ "0": "Continue Conversation",
+ "1": "Pause Conversation",
+ "2": "Restart Conversation",
+ "3": "Repeat Conversation",
+ "4": "End Conversation",
+ "5": "Refuse to Answer",
+ "6": "Affirmative Confirmation",
+ "7": "Negative Confirmation",
+ "8": "Correction Request",
+ "9": "Questions Answerable by Conversational Context / History",
+ "10": "Questions Requiring External Information",
+ "11": "Questions on Process",
+ "12": "New Inquiry or Request",
+ "13": "Transfer to Human Agent"
  },
  "initializer_range": 0.02,
  "label2id": {
- "LABEL_0": 0,
- "LABEL_1": 1,
- "LABEL_10": 10,
- "LABEL_11": 11,
- "LABEL_12": 12,
- "LABEL_13": 13,
- "LABEL_2": 2,
- "LABEL_3": 3,
- "LABEL_4": 4,
- "LABEL_5": 5,
- "LABEL_6": 6,
- "LABEL_7": 7,
- "LABEL_8": 8,
- "LABEL_9": 9
+ "Affirmative Confirmation": 6,
+ "Continue Conversation": 0,
+ "Correction Request": 8,
+ "End Conversation": 4,
+ "Negative Confirmation": 7,
+ "New Inquiry or Request": 12,
+ "Pause Conversation": 1,
+ "Questions Answerable by Conversational Context / History": 9,
+ "Questions Requiring External Information": 10,
+ "Questions on Process": 11,
+ "Refuse to Answer": 5,
+ "Repeat Conversation": 3,
+ "Restart Conversation": 2,
+ "Transfer to Human Agent": 13
  },
  "max_position_embeddings": 512,
  "model_type": "distilbert",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d17797c8e9d7983fb354464411de81f27c877f540742477ea6c08a31cabe5f96
+oid sha256:1aa658f20c80bfa1f5eafca9e90f544823f33f003762c38d7d9ed328d0325b99
 size 267869480