Update labels

config.json

@@ -12,22 +12,22 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1",
-    "2": "LABEL_2",
-    "3": "LABEL_3",
-    "4": "LABEL_4",
-    "5": "LABEL_5"
+    "0": "0",
+    "1": ".",
+    "2": ",",
+    "3": "?",
+    "4": "-",
+    "5": ":"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "LABEL_0": 0,
-    "LABEL_1": 1,
-    "LABEL_2": 2,
-    "LABEL_3": 3,
-    "LABEL_4": 4,
-    "LABEL_5": 5
+    "0": 0,
+    ".": 1,
+    ",": 2,
+    "?": 3,
+    "-": 4,
+    ":": 5
   },
   "layer_norm_eps": 1e-05,
   "max_position_embeddings": 514,

convert-labels.py

@@ -0,0 +1,29 @@
+import shutil
+
+
+if __name__ == "__main__":
+
+    print("Convert model labels to human readable")
+
+    source_file = 'config.json'
+    old_file = 'config.json.bak'
+    
+    shutil.copyfile(source_file, old_file)
+
+    LABEL_TO_CHAR = {
+        "LABEL_0" : "0",
+        "LABEL_1" : ".",
+        "LABEL_2" : ",",
+        "LABEL_3" : "?",
+        "LABEL_4" : "-",
+        "LABEL_5" : ":"
+    }
+
+
+    with open(old_file, 'r') as fp_input, open(source_file, 'w') as fp_output:
+        lines = fp_input.readlines()
+        for line in lines:
+            for label in LABEL_TO_CHAR.keys():
+                line = line.replace(label, LABEL_TO_CHAR[label])
+
+            fp_output.write(line)