Hugging Face's logo

TinyModels
/
Setfit-banking-intent

Text Classification
setfit
Safetensors
sentence-transformers
English
bert
generated_from_setfit_trainer
text-embeddings-inference
Model card Files
xet
 Community
QuantaSparkLabs commited on
Commit
753f67a
·
verified ·
1 Parent(s): a51d900

Push model using huggingface_hub.

Browse files
Files changed (11) hide show
  1. 1_Pooling/config.json +5 -0
  2. README.md +327 -0
  3. config.json +35 -0
  4. config_sentence_transformers.json +14 -0
  5. config_setfit.json +82 -0
  6. model.safetensors +3 -0
  7. model_head.pkl +3 -0
  8. modules.json +20 -0
  9. sentence_bert_config.json +10 -0
  10. tokenizer.json +0 -0
  11. tokenizer_config.json +18 -0
1_Pooling/config.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "embedding_dimension": 384,
3
+ "pooling_mode": "cls",
4
+ "include_prompt": true
5
+ }
README.md ADDED
@@ -0,0 +1,327 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - setfit
4
+ - sentence-transformers
5
+ - text-classification
6
+ - generated_from_setfit_trainer
7
+ widget:
8
+ - text: What do I do? My card is broken.
9
+ - text: Where should I withdraw money from?
10
+ - text: I can't seem to make a standard bank transfer. I have tried at least five
11
+ times already but none of them are going through. Please tell me what is wrong?
12
+ - text: What are the currency exchange fees?
13
+ - text: I have an unauthorized charge.
14
+ metrics:
15
+ - accuracy
16
+ pipeline_tag: text-classification
17
+ library_name: setfit
18
+ inference: true
19
+ base_model: BAAI/bge-small-en-v1.5
20
+ ---
21
+
22
+ # SetFit with BAAI/bge-small-en-v1.5
23
+
24
+ This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. This SetFit model uses [BAAI/bge-small-en-v1.5](https://huggingface.co/BAAI/bge-small-en-v1.5) as the Sentence Transformer embedding model. A [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance is used for classification.
25
+
26
+ The model has been trained using an efficient few-shot learning technique that involves:
27
+
28
+ 1. Fine-tuning a [Sentence Transformer](https://www.sbert.net) with contrastive learning.
29
+ 2. Training a classification head with features from the fine-tuned Sentence Transformer.
30
+
31
+ ## Model Details
32
+
33
+ ### Model Description
34
+ - **Model Type:** SetFit
35
+ - **Sentence Transformer body:** [BAAI/bge-small-en-v1.5](https://huggingface.co/BAAI/bge-small-en-v1.5)
36
+ - **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
37
+ - **Maximum Sequence Length:** 512 tokens
38
+ - **Number of Classes:** 77 classes
39
+ <!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
40
+ <!-- - **Language:** Unknown -->
41
+ <!-- - **License:** Unknown -->
42
+
43
+ ### Model Sources
44
+
45
+ - **Repository:** [SetFit on GitHub](https://github.com/huggingface/setfit)
46
+ - **Paper:** [Efficient Few-Shot Learning Without Prompts](https://arxiv.org/abs/2209.11055)
47
+ - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
48
+
49
+ ### Model Labels
50
+ | Label | Examples |
51
+ |:-------------------------------------------------|:----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
52
+ | activate_my_card | <ul><li>'I have a new card and need to activate it'</li><li>'I would like to have assistance with activating my card.'</li><li>'Would you be so kind as to activate my card? Thanks'</li></ul> |
53
+ | age_limit | <ul><li>'How old do kids have to be to open an account?'</li><li>'At what age can a person open an account?'</li><li>'Am I able to open an account for my children?'</li></ul> |
54
+ | apple_pay_or_google_pay | <ul><li>'Top up is not working even though I have my AMEX in apple pay.'</li><li>'is it possible to top-up with google pay.'</li><li>'How can I get my Google pay top up to work?'</li></ul> |
55
+ | atm_support | <ul><li>'Where should I withdraw money from?'</li><li>'Is there a list of ATMs that take this card?'</li><li>'Which locations can I withdraw money from?'</li></ul> |
56
+ | automatic_top_up | <ul><li>'can you top up your account automatically'</li><li>'How can I setup automatic top-up?'</li><li>'How do I activate auto top-up when traveling?'</li></ul> |
57
+ | balance_not_updated_after_bank_transfer | <ul><li>"my transfer isn't there"</li><li>'I recently transferred some cash, but the balance has not changed.'</li><li>'I just transferred some money into my account and do not see it updated yet, can you assist?'</li></ul> |
58
+ | balance_not_updated_after_cheque_or_cash_deposit | <ul><li>"My account balance doesn't reflect an update I made by cheque yesterday. Is something wrong or is the system slow?"</li><li>"Why isn't my cash deposit showing up in my account?"</li><li>"Cash has been deposited this morning but it hasn't came through."</li></ul> |
59
+ | beneficiary_not_allowed | <ul><li>'I cannot make a transfer to another account.'</li><li>"Why can't I use my beneficiary?"</li><li>'Payment did not process'</li></ul> |
60
+ | cancel_transfer | <ul><li>"Please cancel the transfer I just made. I put in the wrong account number and can't seem to stop it on the app."</li><li>'I need to cancel my recent transfer as soon as possible. I made an error there. Please help before it goes through.'</li><li>'My transaction from earlier needs to be reversed.'</li></ul> |
61
+ | card_about_to_expire | <ul><li>"I'm in China and really need a new card."</li><li>'Do I need to do something to get a new card once it expires?'</li><li>'In a month, my card will reach its expiration date, am I required to request the next card?'</li></ul> |
62
+ | card_acceptance | <ul><li>'Where can I use my Mastercard?'</li><li>'Do all businesses take this card?'</li><li>'Can I use my card no matter where I go?'</li></ul> |
63
+ | card_arrival | <ul><li>'Where is the card I ordered 2 weeks ago?'</li><li>'Can I track when my card will be delivered?'</li><li>"I don't have my card in 1 week. Should I be worried?"</li></ul> |
64
+ | card_delivery_estimate | <ul><li>'Can cards be dispatched on a specific date?'</li><li>'What is the delivery time for US?'</li><li>'Can I pay more to get it deliver quicker?'</li></ul> |
65
+ | card_linking | <ul><li>'My card is no longer lost. How do I enter it back into the app?'</li><li>"My new card isn't in my app, how do I get it in there?"</li><li>'I received my card, how do I get it to work in the app?'</li></ul> |
66
+ | card_not_working | <ul><li>"Why won't my card work."</li><li>'What do I do? My card is broken.'</li><li>'My card will not work'</li></ul> |
67
+ | card_payment_fee_charged | <ul><li>"This fee shouldn't be there after I paid with a card."</li><li>'Why am I getting charged more for using my card?'</li><li>'There was an extra charge when I used my card'</li></ul> |
68
+ | card_payment_not_recognised | <ul><li>"I don't understand where this charge came from."</li><li>'Why is there an uknown charge showing on my account?'</li><li>'The App shows a fraudulent payment made on my account.'</li></ul> |
69
+ | card_payment_wrong_exchange_rate | <ul><li>"There's an error with the rate that is charged."</li><li>'What do i do if the exchange rate was wrong for the item i bought overseas?'</li><li>"My exchange rate isn't correct."</li></ul> |
70
+ | card_swallowed | <ul><li>'What should I do with my stuck ATM?'</li><li>'Please send a new card; the ATM ate mine.'</li><li>'The ATM did not give me my card back?'</li></ul> |
71
+ | cash_withdrawal_charge | <ul><li>'I got charged fees for withdrawing cash!'</li><li>'What is the extra fee I was charged when using an ATM?'</li><li>"I don't understand why I was charged extra for withdrawing money?"</li></ul> |
72
+ | cash_withdrawal_not_recognised | <ul><li>'My app is showing a cash balance that I did not receive.'</li><li>'I am using the app, and I see cash withdrawals that I did not authorize.'</li><li>'Someone has copied my card. I have my card with me, but someone just made a 500£ cash withdrawal. Please help now.'</li></ul> |
73
+ | change_pin | <ul><li>'Can you tell me how to change my PIN?'</li><li>'How can I change my PIN? Help.'</li><li>'Which atms allow me to change my pin?'</li></ul> |
74
+ | compromised_card | <ul><li>'am afraid someone may have gained access to the info on my card.'</li><li>"There are fraudulent charges that I didn't make on my card. Can you please put in the safeguards necessary to stop this from happening anymore?"</li><li>'The card has suffered a security breach.'</li></ul> |
75
+ | contactless_not_working | <ul><li>'How come when I tried to pay contactless at the bus it didnt work?'</li><li>'My contactless stopped working today no matter where I try to use it. How do I get it to start working again?'</li><li>'Why is my contactless not working?'</li></ul> |
76
+ | country_support | <ul><li>'Will my new card work outside of the EU?'</li><li>'Is it possible for a person who lives in the US to get a card?'</li><li>"I'm in the EU. Can I get one of your cards?"</li></ul> |
77
+ | declined_card_payment | <ul><li>'You have declined my payment.'</li><li>'Why do you keep declining my payment?I tried several times already with this card and it is just not working.'</li><li>"My card payment didn't go through."</li></ul> |
78
+ | declined_cash_withdrawal | <ul><li>'I was not able to use the ATM to get cash'</li><li>'Hi, I am unable to use my card in ATM. I had tried my card in two ATMs but both ATMs declined it. Could you please check my account?'</li><li>"Is my card broken? I can't get cash out of the ATM"</li></ul> |
79
+ | declined_transfer | <ul><li>'I got a message that my transfer was declined.'</li><li>'I need to make a transfer but it keeps getting declined I need Help!'</li><li>'I tried to do a transfer but I got a message that it was declined.'</li></ul> |
80
+ | direct_debit_payment_not_recognised | <ul><li>'I just went through a old statement and noticed that a very large payment went through and i did not make it. It was from a few months ago, can I still dispute the payment?'</li><li>'A debit came through that I need to dispute.'</li><li>'I have an unauthorized charge.'</li></ul> |
81
+ | disposable_card_limits | <ul><li>'What are the policies of disposable cards?'</li><li>'I have a disposable virtual card. Is there a limit to how many times I can use it?'</li><li>"Are there any places that I can't use the disposable cards at?"</li></ul> |
82
+ | edit_personal_details | <ul><li>'I just got married and need to change my name on the account'</li><li>"I'd like to make some changes to my personal information."</li><li>'I need to change my last name on my account because I just got married.'</li></ul> |
83
+ | exchange_charge | <ul><li>'Would you be able to offer me a discount, since I need to exchange currencies frequently?'</li><li>'What are the currency exchange fees?'</li><li>'Will there be any extra fees if I exchange currencies?'</li></ul> |
84
+ | exchange_rate | <ul><li>'Where are you getting your exchange rates from?'</li><li>'How are the exchange rates determined?'</li><li>'Is there an exchange rate?'</li></ul> |
85
+ | exchange_via_app | <ul><li>'Can I exchange USD and GBP from the app?'</li><li>'I might need to exchange between USD and GBP, is this possible with your app?'</li><li>'Will it be possible to change from USD to GBP with this App?'</li></ul> |
86
+ | extra_charge_on_statement | <ul><li>'How come I was charged a transaction of $1?'</li><li>'What is this fee debit on my account?'</li><li>'Why is there a $1 charge on my statement?'</li></ul> |
87
+ | failed_transfer | <ul><li>"I can't seem to make a standard bank transfer. I have tried at least five times already but none of them are going through. Please tell me what is wrong?"</li><li>"Hi, I'm trying to purchase a flat and I'm needing the first payment for the mortgage to go through on the flat I'm trying to buy. For some reason I keep getting an error.can you assist in making this transfer? What is going on?"</li><li>'Did my transfer fail?'</li></ul> |
88
+ | fiat_currency_support | <ul><li>'How many currencies can I have?'</li><li>'What are eligible holding and exchange currencies?'</li><li>'What currencies can I not hold many in, if so what are they?'</li></ul> |
89
+ | get_disposable_virtual_card | <ul><li>'What can you use the disposable cards for?'</li><li>'Can you tell me where I can order a disposable virtual card?'</li><li>"Can you point me in the right direction of where to go to roder a disposable virtual card, I can't seem to find it."</li></ul> |
90
+ | get_physical_card | <ul><li>"I havn't received my PIN yet."</li><li>'How long do I wait for my PIN'</li><li>'In the app, where do I find my card PIN?'</li></ul> |
91
+ | getting_spare_card | <ul><li>'I would like to get a few more physical cards.'</li><li>'Do you charge for sending more cards?'</li><li>'Is it possible to get another card?'</li></ul> |
92
+ | getting_virtual_card | <ul><li>'How do I receive a virtual card?'</li><li>'How can I sign up for a virtual card?'</li><li>'Why have I not received a virtual card yet?'</li></ul> |
93
+ | lost_or_stolen_card | <ul><li>'I think my card is lost. What do I do?'</li><li>'Can you tell me the status of my new card?'</li><li>"I've lost my card. What can I do about that?"</li></ul> |
94
+ | lost_or_stolen_phone | <ul><li>'My app was on the phone and I was mugged. What do I do?'</li><li>'Can my account me accessed by a thief? I was mugged.'</li><li>"I don't know where my phone is."</li></ul> |
95
+ | order_physical_card | <ul><li>'Can I get a card in person?'</li><li>'tell me how I can order my card.'</li><li>'how do I ask for a physical card?'</li></ul> |
96
+ | passcode_forgotten | <ul><li>'How can I change my password?'</li><li>'I thought I knew my password but I guess I was wrong, what can I do now?'</li><li>'What are the steps for changing my passcode?'</li></ul> |
97
+ | pending_card_payment | <ul><li>'A card payment on my account is shown as pending.'</li><li>'I used my card to make a payment a while ago and its still pending, Why is it taking so long?'</li><li>"My payment is in the app but hasn't gone through yet"</li></ul> |
98
+ | pending_cash_withdrawal | <ul><li>"I tried to get some with-drawls but the machine didn't work. The transaction still seems in progress.Seems like something is wrong,I don't want to be charged for Transaction i did not make."</li><li>'Why would a cash withdrawal still be pending?'</li><li>'My ATM withdrawal is taking forever'</li></ul> |
99
+ | pending_top_up | <ul><li>'What is taking so long with my top up?'</li><li>'Why is the top-up I made still pending.'</li><li>'What do I do if my top-up has not gone through?'</li></ul> |
100
+ | pending_transfer | <ul><li>'The wait for this transfer seems to be taking a very long time. It is very important that it goes through quickly. Can you please tell me how long it may take?'</li><li>"Hi, I'm trying to transfer money from one country to another, but it's taking a long time! All the account information is right, so why is it taking so long?"</li><li>'I am still waiting for a transfer to show up'</li></ul> |
101
+ | pin_blocked | <ul><li>'I blocked my card by mistake, how do I unblock it?'</li><li>'How come I got blocked for entering wrong pins?'</li><li>'Help me unblock my account. I entered the PIN wrong too many times.'</li></ul> |
102
+ | receiving_money | <ul><li>'Can friends transfer money to each other?'</li><li>'Do I need to do anything special for my friend to send money to my account?'</li><li>'How can my boss pay me directly to the card?'</li></ul> |
103
+ | Refund_not_showing_up | <ul><li>'How do I find the refund on my statement?'</li><li>'My refund has not arrived, and I am unsure why. Who do I contact?'</li><li>"Why can't I see my refund in my statement?"</li></ul> |
104
+ | request_refund | <ul><li>'I want to cancel a purchase'</li><li>'Can I request a refund'</li><li>'I need a refund for something I bought.'</li></ul> |
105
+ | reverted_card_payment? | <ul><li>'I tried to buy something online but my funds were transferred back to my account. What is happening?'</li><li>'I did a payment but it was reverted by the app'</li><li>"I purchased something about one to two weeks ago but for some reason the payment was returned to my account today even though I've already received one of the items. Can you tell me what's going on?"</li></ul> |
106
+ | supported_cards_and_currencies | <ul><li>'Can I use this app for any currency?'</li><li>'Can you tell me what currencies I can use to add money?'</li><li>'What currencies are approved to add money?'</li></ul> |
107
+ | terminate_account | <ul><li>'I wish to remove my account.'</li><li>'How can I delete my account?'</li><li>'I want to deactivate my account.'</li></ul> |
108
+ | top_up_by_bank_transfer_charge | <ul><li>'How much does it cost to do a transfer?'</li><li>'If I top up by transfer, am I going to be charged?'</li><li>'Is there a cost to receive money?'</li></ul> |
109
+ | top_up_by_card_charge | <ul><li>'How much does it cost to top up a US-issued card?'</li><li>'Do you have any fees if I want to add money using an international card.'</li><li>'I want to add money to my international card, so are there any fees charged to do this?'</li></ul> |
110
+ | top_up_by_cash_or_cheque | <ul><li>'can i top up with check?'</li><li>'can i deposit a check here?'</li><li>'i would like to top up with check'</li></ul> |
111
+ | top_up_failed | <ul><li>"I don't think that my top-up worked."</li><li>"Won't let me top my card up!! Who can get it sorted out for me? It's urgent! How do I find out what the problem is with my card and how else to top it up?"</li><li>"I am having a huge problem with my card being declined and it's really important that I top-up today. Would you please look into this further and let me know how I can get this fixed?"</li></ul> |
112
+ | top_up_limits | <ul><li>'Can I use as many top-ups as I want?'</li><li>"What's the top-up limit?"</li><li>'Can I create my own top-up limit?'</li></ul> |
113
+ | top_up_reverted | <ul><li>'Why did my top-up get reverted?'</li><li>"Why didn't my top up work?"</li><li>'What if my top up is reverted?'</li></ul> |
114
+ | topping_up_by_card | <ul><li>'my top up isnt there yet where do i find it'</li><li>"I can't see my top up in my wallet!"</li><li>'Is it possible to transfer money to my credit card?'</li></ul> |
115
+ | transaction_charged_twice | <ul><li>'I have a duplicate charge'</li><li>'What happens with the duplicate charge in my account?'</li><li>"What do I do if I've been charged twice for the same transaction?"</li></ul> |
116
+ | transfer_fee_charged | <ul><li>'I make payments for my place abroad using this account and since i use it so frequently I was wondering why did the fees increase suddenly rather than decreasing? I would think customers like me should receive incentives for using your service'</li><li>'Why was I charged a fee for transferring money'</li><li>'I was charged a fee for the transfer, but can you tell me why?'</li></ul> |
117
+ | transfer_into_account | <ul><li>'I want to transfer a sum of money from another bank account I have into this one'</li><li>'How can I transfer money to this account from another bank?'</li><li>'How does your app transfer the money'</li></ul> |
118
+ | transfer_not_received_by_recipient | <ul><li>"I transferred money and it didn't get there."</li><li>"I'm still expecting the transaction to be finished"</li><li>'I am having an issue with an in country transfer I did a few days ago. It has yet to appear in my account. I have checked to be sure that all account information is correct multiple times. What is taking the transfer so long?'</li></ul> |
119
+ | transfer_timing | <ul><li>'When will the money be made available in my account?'</li><li>'What is the length of time for money to reach the account?'</li><li>'How long until transfers from Europe go through?'</li></ul> |
120
+ | unable_to_verify_identity | <ul><li>'The app is not able to realize that it is me.'</li><li>'why does it take so long to verify my id'</li><li>'Hearing about your verification results from us may take anywhere from 10 minutes to approximately one hour. If this verification has in fact, failed, double-check to make sure that your images are clear. Also make sure that your document photos have no blur or glare. They need to be readable. You must also be 18 years of age or older and be a resident of Switzerland or the European Economic Area to open an account.'</li></ul> |
121
+ | verify_my_identity | <ul><li>'Can I get a full list of instructions for identity checks?'</li><li>'How recent do my documents need to be for verification?'</li><li>'Can I get information on the identity checks?'</li></ul> |
122
+ | verify_source_of_funds | <ul><li>'I would like to know how to work out where my funds came from?'</li><li>"I'd like to know where my funds come from."</li><li>'how do i track the moeney'</li></ul> |
123
+ | verify_top_up | <ul><li>'What is the significance of verifying the top-up?'</li><li>'How are top-ups verified?'</li><li>"I don't see the verification code for my card transfer."</li></ul> |
124
+ | virtual_card_not_working | <ul><li>'My disposable virtual card got rejected'</li><li>"I can't get the virtual card to work"</li><li>"Why isn't my disposable virtual card working?"</li></ul> |
125
+ | visa_or_mastercard | <ul><li>'Which one are you? Visa or Mastercard?'</li><li>'Is it possible to get both Mastercard and Visa card from you?'</li><li>'is it ok if i use both visa and mastercard'</li></ul> |
126
+ | why_verify_identity | <ul><li>'What is the function of the identity check'</li><li>'I am not comfortable verifying my indentity.'</li><li>"My ID verification hasn't gone through yet, but can I use my account anyway?"</li></ul> |
127
+ | wrong_amount_of_cash_received | <ul><li>'I only got $20 of the $100 that I attempted to withdraw.'</li><li>'Got only $20 when I tried to withdraw $100.'</li><li>'I received less cash than the amount I specified in the ATM.'</li></ul> |
128
+ | wrong_exchange_rate_for_cash_withdrawal | <ul><li>'what is the correct exchange rate for getting cash'</li><li>'On my last trip, I received the incorrect exchange rate.'</li><li>'I used the ATM to get money out and did not realize I would be charged as much as I was or I would not have done that!'</li></ul> |
129
+
130
+ ## Uses
131
+
132
+ ### Direct Use for Inference
133
+
134
+ First install the SetFit library:
135
+
136
+ ```bash
137
+ pip install setfit
138
+ ```
139
+
140
+ Then you can load this model and run inference.
141
+
142
+ ```python
143
+ from setfit import SetFitModel
144
+
145
+ # Download from the 🤗 Hub
146
+ model = SetFitModel.from_pretrained("QuantaSparkLabs/setfit-banking-intent")
147
+ # Run inference
148
+ preds = model("I have an unauthorized charge.")
149
+ ```
150
+
151
+ <!--
152
+ ### Downstream Use
153
+
154
+ *List how someone could finetune this model on their own dataset.*
155
+ -->
156
+
157
+ <!--
158
+ ### Out-of-Scope Use
159
+
160
+ *List how the model may foreseeably be misused and address what users ought not to do with the model.*
161
+ -->
162
+
163
+ <!--
164
+ ## Bias, Risks and Limitations
165
+
166
+ *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
167
+ -->
168
+
169
+ <!--
170
+ ### Recommendations
171
+
172
+ *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
173
+ -->
174
+
175
+ ## Training Details
176
+
177
+ ### Training Set Metrics
178
+ | Training set | Min | Median | Max |
179
+ |:-------------|:----|:--------|:----|
180
+ | Word count | 4 | 11.6623 | 78 |
181
+
182
+ | Label | Training Sample Count |
183
+ |:-------------------------------------------------|:----------------------|
184
+ | activate_my_card | 4 |
185
+ | age_limit | 4 |
186
+ | apple_pay_or_google_pay | 4 |
187
+ | atm_support | 4 |
188
+ | automatic_top_up | 4 |
189
+ | balance_not_updated_after_bank_transfer | 4 |
190
+ | balance_not_updated_after_cheque_or_cash_deposit | 4 |
191
+ | beneficiary_not_allowed | 4 |
192
+ | cancel_transfer | 4 |
193
+ | card_about_to_expire | 4 |
194
+ | card_acceptance | 4 |
195
+ | card_arrival | 4 |
196
+ | card_delivery_estimate | 4 |
197
+ | card_linking | 4 |
198
+ | card_not_working | 4 |
199
+ | card_payment_fee_charged | 4 |
200
+ | card_payment_not_recognised | 4 |
201
+ | card_payment_wrong_exchange_rate | 4 |
202
+ | card_swallowed | 4 |
203
+ | cash_withdrawal_charge | 4 |
204
+ | cash_withdrawal_not_recognised | 4 |
205
+ | change_pin | 4 |
206
+ | compromised_card | 4 |
207
+ | contactless_not_working | 4 |
208
+ | country_support | 4 |
209
+ | declined_card_payment | 4 |
210
+ | declined_cash_withdrawal | 4 |
211
+ | declined_transfer | 4 |
212
+ | direct_debit_payment_not_recognised | 4 |
213
+ | disposable_card_limits | 4 |
214
+ | edit_personal_details | 4 |
215
+ | exchange_charge | 4 |
216
+ | exchange_rate | 4 |
217
+ | exchange_via_app | 4 |
218
+ | extra_charge_on_statement | 4 |
219
+ | failed_transfer | 4 |
220
+ | fiat_currency_support | 4 |
221
+ | get_disposable_virtual_card | 4 |
222
+ | get_physical_card | 4 |
223
+ | getting_spare_card | 4 |
224
+ | getting_virtual_card | 4 |
225
+ | lost_or_stolen_card | 4 |
226
+ | lost_or_stolen_phone | 4 |
227
+ | order_physical_card | 4 |
228
+ | passcode_forgotten | 4 |
229
+ | pending_card_payment | 4 |
230
+ | pending_cash_withdrawal | 4 |
231
+ | pending_top_up | 4 |
232
+ | pending_transfer | 4 |
233
+ | pin_blocked | 4 |
234
+ | receiving_money | 4 |
235
+ | Refund_not_showing_up | 4 |
236
+ | request_refund | 4 |
237
+ | reverted_card_payment? | 4 |
238
+ | supported_cards_and_currencies | 4 |
239
+ | terminate_account | 4 |
240
+ | top_up_by_bank_transfer_charge | 4 |
241
+ | top_up_by_card_charge | 4 |
242
+ | top_up_by_cash_or_cheque | 4 |
243
+ | top_up_failed | 4 |
244
+ | top_up_limits | 4 |
245
+ | top_up_reverted | 4 |
246
+ | topping_up_by_card | 4 |
247
+ | transaction_charged_twice | 4 |
248
+ | transfer_fee_charged | 4 |
249
+ | transfer_into_account | 4 |
250
+ | transfer_not_received_by_recipient | 4 |
251
+ | transfer_timing | 4 |
252
+ | unable_to_verify_identity | 4 |
253
+ | verify_my_identity | 4 |
254
+ | verify_source_of_funds | 4 |
255
+ | verify_top_up | 4 |
256
+ | virtual_card_not_working | 4 |
257
+ | visa_or_mastercard | 4 |
258
+ | why_verify_identity | 4 |
259
+ | wrong_amount_of_cash_received | 4 |
260
+ | wrong_exchange_rate_for_cash_withdrawal | 4 |
261
+
262
+ ### Training Hyperparameters
263
+ - batch_size: (32, 32)
264
+ - num_epochs: (1, 1)
265
+ - max_steps: -1
266
+ - sampling_strategy: oversampling
267
+ - num_iterations: 2
268
+ - body_learning_rate: (2e-05, 2e-05)
269
+ - head_learning_rate: 2e-05
270
+ - loss: CosineSimilarityLoss
271
+ - distance_metric: cosine_distance
272
+ - margin: 0.25
273
+ - end_to_end: False
274
+ - use_amp: False
275
+ - warmup_proportion: 0.1
276
+ - l2_weight: 0.01
277
+ - seed: 42
278
+ - eval_max_steps: -1
279
+ - load_best_model_at_end: False
280
+
281
+ ### Training Results
282
+ | Epoch | Step | Training Loss | Validation Loss |
283
+ |:------:|:----:|:-------------:|:---------------:|
284
+ | 0.0256 | 1 | 0.2141 | - |
285
+
286
+ ### Framework Versions
287
+ - Python: 3.13.15
288
+ - SetFit: 1.2.0
289
+ - Sentence Transformers: 5.7.0
290
+ - Transformers: 5.16.1
291
+ - PyTorch: 2.11.0+cu128
292
+ - Datasets: 5.0.1
293
+ - Tokenizers: 0.23.1
294
+
295
+ ## Citation
296
+
297
+ ### BibTeX
298
+ ```bibtex
299
+ @article{https://doi.org/10.48550/arxiv.2209.11055,
300
+ doi = {10.48550/ARXIV.2209.11055},
301
+ url = {https://arxiv.org/abs/2209.11055},
302
+ author = {Tunstall, Lewis and Reimers, Nils and Jo, Unso Eun Seo and Bates, Luke and Korat, Daniel and Wasserblat, Moshe and Pereg, Oren},
303
+ keywords = {Computation and Language (cs.CL), FOS: Computer and information sciences, FOS: Computer and information sciences},
304
+ title = {Efficient Few-Shot Learning Without Prompts},
305
+ publisher = {arXiv},
306
+ year = {2022},
307
+ copyright = {Creative Commons Attribution 4.0 International}
308
+ }
309
+ ```
310
+
311
+ <!--
312
+ ## Glossary
313
+
314
+ *Clearly define terms in order to be accessible across audiences.*
315
+ -->
316
+
317
+ <!--
318
+ ## Model Card Authors
319
+
320
+ *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
321
+ -->
322
+
323
+ <!--
324
+ ## Model Card Contact
325
+
326
+ *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
327
+ -->
config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_cross_attention": false,
3
+ "architectures": [
4
+ "BertModel"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": null,
8
+ "classifier_dropout": null,
9
+ "dtype": "float32",
10
+ "eos_token_id": null,
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.1,
13
+ "hidden_size": 384,
14
+ "id2label": {
15
+ "0": "LABEL_0"
16
+ },
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 1536,
19
+ "is_decoder": false,
20
+ "label2id": {
21
+ "LABEL_0": 0
22
+ },
23
+ "layer_norm_eps": 1e-12,
24
+ "max_position_embeddings": 512,
25
+ "model_type": "bert",
26
+ "num_attention_heads": 12,
27
+ "num_hidden_layers": 12,
28
+ "pad_token_id": 0,
29
+ "position_embedding_type": "absolute",
30
+ "tie_word_embeddings": true,
31
+ "transformers_version": "5.16.1",
32
+ "type_vocab_size": 2,
33
+ "use_cache": false,
34
+ "vocab_size": 30522
35
+ }
config_sentence_transformers.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "__version__": {
3
+ "pytorch": "2.11.0+cu128",
4
+ "sentence_transformers": "5.7.0",
5
+ "transformers": "5.16.1"
6
+ },
7
+ "default_prompt_name": null,
8
+ "model_type": "SentenceTransformer",
9
+ "prompts": {
10
+ "document": "",
11
+ "query": ""
12
+ },
13
+ "similarity_fn_name": "cosine"
14
+ }
config_setfit.json ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "labels": [
3
+ "activate_my_card",
4
+ "age_limit",
5
+ "apple_pay_or_google_pay",
6
+ "atm_support",
7
+ "automatic_top_up",
8
+ "balance_not_updated_after_bank_transfer",
9
+ "balance_not_updated_after_cheque_or_cash_deposit",
10
+ "beneficiary_not_allowed",
11
+ "cancel_transfer",
12
+ "card_about_to_expire",
13
+ "card_acceptance",
14
+ "card_arrival",
15
+ "card_delivery_estimate",
16
+ "card_linking",
17
+ "card_not_working",
18
+ "card_payment_fee_charged",
19
+ "card_payment_not_recognised",
20
+ "card_payment_wrong_exchange_rate",
21
+ "card_swallowed",
22
+ "cash_withdrawal_charge",
23
+ "cash_withdrawal_not_recognised",
24
+ "change_pin",
25
+ "compromised_card",
26
+ "contactless_not_working",
27
+ "country_support",
28
+ "declined_card_payment",
29
+ "declined_cash_withdrawal",
30
+ "declined_transfer",
31
+ "direct_debit_payment_not_recognised",
32
+ "disposable_card_limits",
33
+ "edit_personal_details",
34
+ "exchange_charge",
35
+ "exchange_rate",
36
+ "exchange_via_app",
37
+ "extra_charge_on_statement",
38
+ "failed_transfer",
39
+ "fiat_currency_support",
40
+ "get_disposable_virtual_card",
41
+ "get_physical_card",
42
+ "getting_spare_card",
43
+ "getting_virtual_card",
44
+ "lost_or_stolen_card",
45
+ "lost_or_stolen_phone",
46
+ "order_physical_card",
47
+ "passcode_forgotten",
48
+ "pending_card_payment",
49
+ "pending_cash_withdrawal",
50
+ "pending_top_up",
51
+ "pending_transfer",
52
+ "pin_blocked",
53
+ "receiving_money",
54
+ "Refund_not_showing_up",
55
+ "request_refund",
56
+ "reverted_card_payment?",
57
+ "supported_cards_and_currencies",
58
+ "terminate_account",
59
+ "top_up_by_bank_transfer_charge",
60
+ "top_up_by_card_charge",
61
+ "top_up_by_cash_or_cheque",
62
+ "top_up_failed",
63
+ "top_up_limits",
64
+ "top_up_reverted",
65
+ "topping_up_by_card",
66
+ "transaction_charged_twice",
67
+ "transfer_fee_charged",
68
+ "transfer_into_account",
69
+ "transfer_not_received_by_recipient",
70
+ "transfer_timing",
71
+ "unable_to_verify_identity",
72
+ "verify_my_identity",
73
+ "verify_source_of_funds",
74
+ "verify_top_up",
75
+ "virtual_card_not_working",
76
+ "visa_or_mastercard",
77
+ "why_verify_identity",
78
+ "wrong_amount_of_cash_received",
79
+ "wrong_exchange_rate_for_cash_withdrawal"
80
+ ],
81
+ "normalize_embeddings": false
82
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03326727087df900f37a84e94b195ddcf74601643eac133da0795af988af868c
3
+ size 133462128
model_head.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6bf8498dc61c8c5b50323ddc1db946059b25288d0cb994f87964a91f4bdfd71e
3
+ size 120059
modules.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "idx": 0,
4
+ "name": "0",
5
+ "path": "",
6
+ "type": "sentence_transformers.base.modules.transformer.Transformer"
7
+ },
8
+ {
9
+ "idx": 1,
10
+ "name": "1",
11
+ "path": "1_Pooling",
12
+ "type": "sentence_transformers.sentence_transformer.modules.pooling.Pooling"
13
+ },
14
+ {
15
+ "idx": 2,
16
+ "name": "2",
17
+ "path": "2_Normalize",
18
+ "type": "sentence_transformers.sentence_transformer.modules.normalize.Normalize"
19
+ }
20
+ ]
sentence_bert_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "transformer_task": "feature-extraction",
3
+ "modality_config": {
4
+ "text": {
5
+ "method": "forward",
6
+ "method_output_name": "last_hidden_state"
7
+ }
8
+ },
9
+ "module_output_name": "token_embeddings"
10
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": "tokenizers",
3
+ "clean_up_tokenization_spaces": true,
4
+ "cls_token": "[CLS]",
5
+ "do_basic_tokenize": true,
6
+ "do_lower_case": true,
7
+ "is_local": false,
8
+ "local_files_only": false,
9
+ "mask_token": "[MASK]",
10
+ "model_max_length": 512,
11
+ "never_split": null,
12
+ "pad_token": "[PAD]",
13
+ "sep_token": "[SEP]",
14
+ "strip_accents": null,
15
+ "tokenize_chinese_chars": true,
16
+ "tokenizer_class": "BertTokenizer",
17
+ "unk_token": "[UNK]"
18
+ }