Downloads · 30 days
26
1% of all-time downloads
roberta-large-vira-intents
ibm-research/roberta-large-vira-intents
roberta-large-vira-intents is a text classification model from ibm-research. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as other.
This model is based on RoBERTa large (Liu, 2019), fine-tuned on a dataset of intent expressions available here and also on 🤗 Transformer datasets hub here.
All-time downloads
2.2K
Public
Repo size
4.3 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin1.4 GB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- other
- Model type
- roberta
- Access
- Public
- Created
- May 31, 2022
- Updated
- Jun 1, 2022
- SHA
- f834137c
Try a prompt
Datasets
Model card
From the Hugging Face model README
Model Description
This model is based on RoBERTa large (Liu, 2019), fine-tuned on a dataset of intent expressions available here and also on 🤗 Transformer datasets hub here.
The model was created as part of the work described in Benchmark Data and Evaluation Framework for Intent Discovery Around COVID-19 Vaccine Hesitancy . The model is released under the Community Data License Agreement - Sharing - Version 1.0 (link), If you use this model, please cite our paper.
The official GitHub is here. The script used for training the model is trainer.py.
Training parameters
- base_model = 'roberta-large'
- learning_rate=5e-6
- per_device_train_batch_size=16,
- per_device_eval_batch_size=16,
- num_train_epochs=15,
- load_best_model_at_end=True,
- save_total_limit=1,
- save_strategy='epoch',
- evaluation_strategy='epoch',
- metric_for_best_model='accuracy',
- seed=123
Data collator
DataCollatorWithPadding
- Task
- Text Classification
- Library
- transformers
- Type
- roberta
- License
- other
- Languages
- en
- Created
- May 31, 2022
- Updated
- Jun 1, 2022