Update README.md

README.md

@@ -1,4 +1,19 @@
 ---
+tokenizer:
+  name_or_path: bert-base-uncased  # Replace with your preferred tokenizer, or use the same as the one used in training
+
+task_specific:
+  text_classification:
+    num_labels: 3  # Adjust based on the number of categories in your classification task
+    label_stoi:
+      CLASSIFY: 0
+      POSITIVE: 1
+      NEGATIVE: 2
+    label_itos:
+      0: CLASSIFY
+      1: POSITIVE
+      2: NEGATIVE
+    threshold: 0.5  # Adjust based on your desired probability threshold for label assignment
 language: en
 tags:
 - exbert