Update README.md
Browse files
README.md
CHANGED
|
@@ -24,6 +24,19 @@ We have finetuned Base Bert model for text classification task. We used intent-d
|
|
| 24 |
|
| 25 |
More information needed
|
| 26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
## Training and evaluation data
|
| 28 |
|
| 29 |
More information needed
|
|
|
|
| 24 |
|
| 25 |
More information needed
|
| 26 |
|
| 27 |
+
## How to use
|
| 28 |
+
|
| 29 |
+
Use below code to test the model
|
| 30 |
+
|
| 31 |
+
new_model = AutoModelForSequenceClassification.from_pretrained("ArunAIML/bert-model-intent-classification",
|
| 32 |
+
num_labels=21,
|
| 33 |
+
id2label=id_to_label,
|
| 34 |
+
label2id=label_to_id)
|
| 35 |
+
|
| 36 |
+
tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
|
| 37 |
+
|
| 38 |
+
|
| 39 |
+
|
| 40 |
## Training and evaluation data
|
| 41 |
|
| 42 |
More information needed
|