kairaamilanii
commited on
Commit
•
50a4867
1
Parent(s):
7d509cc
Update README.md
Browse files
README.md
CHANGED
@@ -52,7 +52,28 @@ model-index:
|
|
52 |
type: f1
|
53 |
value: 0.9724
|
54 |
---
|
|
|
55 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
56 |
```python
|
57 |
>>> from transformers import pipeline
|
58 |
|
@@ -63,4 +84,8 @@ text = "hi what's the maximum amount of money I can withdraw from" # Replace wit
|
|
63 |
|
64 |
prediction = classifier(text)
|
65 |
prediction
|
|
|
|
|
|
|
|
|
66 |
```
|
|
|
52 |
type: f1
|
53 |
value: 0.9724
|
54 |
---
|
55 |
+
this model based on roberta model that trained with minds-14 dataset, only trained in english version : enUS + enAU + enGB
|
56 |
|
57 |
+
the intent_classes available:
|
58 |
+
```python
|
59 |
+
intent_classes = {
|
60 |
+
0: 'abroad',
|
61 |
+
1: 'address',
|
62 |
+
2: 'app_error',
|
63 |
+
3: 'atm_limit',
|
64 |
+
4: 'balance',
|
65 |
+
5: 'business_loan',
|
66 |
+
6: 'card_issues',
|
67 |
+
7: 'cash_deposit',
|
68 |
+
8: 'direct_debit',
|
69 |
+
9: 'freeze',
|
70 |
+
10: 'high_value_payment',
|
71 |
+
11: 'joint_account',
|
72 |
+
12: 'latest_transactions',
|
73 |
+
13: 'pay_bill'
|
74 |
+
}
|
75 |
+
```
|
76 |
+
example of using model to classify intent:
|
77 |
```python
|
78 |
>>> from transformers import pipeline
|
79 |
|
|
|
84 |
|
85 |
prediction = classifier(text)
|
86 |
prediction
|
87 |
+
```
|
88 |
+
example output:
|
89 |
+
```python
|
90 |
+
[{'label': 'LABEL_3', 'score': 0.9933607578277588}]
|
91 |
```
|