ArunAIML commited on
Commit
c090765
·
verified ·
1 Parent(s): 4f1cf3a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +35 -1
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
- More information needed
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