kairaamilanii commited on
Commit
50a4867
1 Parent(s): 7d509cc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +25 -0
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
  ```