sidde commited on
Commit
e54703b
·
verified ·
1 Parent(s): bf64a1d

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +67 -0
README.md ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: apache-2.0
4
+ tags:
5
+ - text-classification
6
+ - intent-detection
7
+ - distilbert
8
+ - nlu
9
+ datasets:
10
+ - custom
11
+ metrics:
12
+ - accuracy
13
+ - f1
14
+ pipeline_tag: text-classification
15
+ ---
16
+
17
+ # DistilBERT NLU Intent Classification
18
+
19
+ Fine-tuned DistilBERT model for intent classification in Natural Language Understanding (NLU) systems.
20
+
21
+ ## Model Details
22
+
23
+ - **Base Model:** distilbert-base-uncased
24
+ - **Task:** Intent Classification (Sequence Classification)
25
+ - **Number of Labels:** 8
26
+ - **Framework:** PyTorch + Transformers
27
+
28
+ ## Supported Intents
29
+
30
+ | ID | Intent | Description |
31
+ |----|--------|-------------|
32
+ | 0 | BILLING_ISSUE | Problems with bills or charges |
33
+ | 1 | CANCEL_SUBSCRIPTION | Cancel service requests |
34
+ | 2 | CHECK_BALANCE | Balance inquiry |
35
+ | 3 | GOODBYE | Farewell messages |
36
+ | 4 | GREETING | Hello/welcome messages |
37
+ | 5 | MODIFY_PROFILE | Update account details |
38
+ | 6 | ROAMING_ACTIVATION | Enable roaming |
39
+ | 7 | ROAMING_DEACTIVATION | Disable roaming |
40
+
41
+ ## Usage
42
+
43
+ from transformers import pipeline
44
+
45
+ classifier = pipeline("text-classification", model="sidde/distilbert-nlu-intent-classification")
46
+
47
+ # Single prediction
48
+ result = classifier("I want to check my balance")
49
+ print(result)
50
+ # [{"label": "CHECK_BALANCE", "score": 0.98}]
51
+
52
+ ## Training Details
53
+
54
+ - **Dataset:** 772 examples (custom intent dataset)
55
+ - **Train/Eval Split:** 80/20 with stratification
56
+ - **Epochs:** 10
57
+ - **Batch Size:** 16
58
+ - **Learning Rate:** 2e-5
59
+ - **Hardware:** NVIDIA L4 GPU on OpenShift AI
60
+
61
+ ## Deployment
62
+
63
+ This model is deployed on OpenShift AI using KServe.
64
+
65
+ ## License
66
+
67
+ Apache 2.0