Update README.md
Browse files
README.md
CHANGED
|
@@ -18,7 +18,7 @@ This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co
|
|
| 18 |
|
| 19 |
## Model description
|
| 20 |
|
| 21 |
-
|
| 22 |
|
| 23 |
## Intended uses & limitations
|
| 24 |
|
|
@@ -51,3 +51,37 @@ The following hyperparameters were used during training:
|
|
| 51 |
- Pytorch 2.6.0+cu124
|
| 52 |
- Datasets 2.14.4
|
| 53 |
- Tokenizers 0.21.1
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
|
| 19 |
## Model description
|
| 20 |
|
| 21 |
+
We have finetuned Base Bert model for text classification task. We used intent-detection dataset for traning our model.
|
| 22 |
|
| 23 |
## Intended uses & limitations
|
| 24 |
|
|
|
|
| 51 |
- Pytorch 2.6.0+cu124
|
| 52 |
- Datasets 2.14.4
|
| 53 |
- Tokenizers 0.21.1
|
| 54 |
+
|
| 55 |
+
### Results
|
| 56 |
+
|
| 57 |
+
|
| 58 |
+
Generating detailed evaluation metrics on the validation set...
|
| 59 |
+
|
| 60 |
+
Classification Report:
|
| 61 |
+
precision recall f1-score support
|
| 62 |
+
|
| 63 |
+
100_NIGHT_TRIAL_OFFER 1.00 1.00 1.00 4
|
| 64 |
+
ABOUT_SOF_MATTRESS 1.00 1.00 1.00 2
|
| 65 |
+
CANCEL_ORDER 1.00 1.00 1.00 2
|
| 66 |
+
CHECK_PINCODE 1.00 1.00 1.00 2
|
| 67 |
+
COD 1.00 1.00 1.00 2
|
| 68 |
+
COMPARISON 0.33 0.50 0.40 2
|
| 69 |
+
DELAY_IN_DELIVERY 1.00 1.00 1.00 2
|
| 70 |
+
DISTRIBUTORS 1.00 1.00 1.00 7
|
| 71 |
+
EMI 1.00 1.00 1.00 5
|
| 72 |
+
ERGO_FEATURES 1.00 1.00 1.00 2
|
| 73 |
+
LEAD_GEN 1.00 1.00 1.00 4
|
| 74 |
+
MATTRESS_COST 1.00 0.80 0.89 5
|
| 75 |
+
OFFERS 1.00 1.00 1.00 2
|
| 76 |
+
ORDER_STATUS 1.00 0.75 0.86 4
|
| 77 |
+
ORTHO_FEATURES 1.00 1.00 1.00 4
|
| 78 |
+
PILLOWS 1.00 1.00 1.00 2
|
| 79 |
+
PRODUCT_VARIANTS 0.50 0.50 0.50 4
|
| 80 |
+
RETURN_EXCHANGE 1.00 0.67 0.80 3
|
| 81 |
+
SIZE_CUSTOMIZATION 0.50 0.50 0.50 2
|
| 82 |
+
WARRANTY 0.67 1.00 0.80 2
|
| 83 |
+
WHAT_SIZE_TO_ORDER 0.80 1.00 0.89 4
|
| 84 |
+
|
| 85 |
+
accuracy 0.89 66
|
| 86 |
+
macro avg 0.90 0.89 0.89 66
|
| 87 |
+
weighted avg 0.91 0.89 0.90 66
|