Downloads · 30 days
22
40% of all-time downloads
kairaamilanii/RoBERTa-minds14-en
RoBERTa-minds14-en is a text classification model from kairaamilanii. Use it when you need a label for a piece of text. The card lists the license as unknown.
this model based on roberta model that trained with minds-14 dataset, only trained in english version : enUS + enAU + enGB
Downloads · 30 days
22
40% of all-time downloads
All-time downloads
55
Public
Parameters
125M
499 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors499 MB · 99%
From the Hugging Face model README
this model based on roberta model that trained with minds-14 dataset, only trained in english version : enUS + enAU + enGB
the intent_classes available:
intent_classes = {
0: 'abroad',
1: 'address',
2: 'app_error',
3: 'atm_limit',
4: 'balance',
5: 'business_loan',
6: 'card_issues',
7: 'cash_deposit',
8: 'direct_debit',
9: 'freeze',
10: 'high_value_payment',
11: 'joint_account',
12: 'latest_transactions',
13: 'pay_bill'
}
example of using model to classify intent:
>>> from transformers import pipeline
model = "/content/RoBERTa-mind14-classifier-intent"
classifier = pipeline("text-classification", model=model)
text = "hi what's the maximum amount of money I can withdraw from" # Replace with your desired input text
prediction = classifier(text)
prediction