Downloads · 30 days
9
2% of all-time downloads
roberta-base-cookdial-v1_1
kdudzic/roberta-base-cookdial-v1_1
roberta-base-cookdial-v1_1 is a text classification model from kdudzic. Use it when you need a label for a piece of text. It is set up for transformers.
- Baseline NLU model for the "AMUseBot" cooking taskbot prototype. Updated version with more robust reqingredient intent recognition thanks to finetuning with extra synthetic data. - roberta-base model finetuned with…
All-time downloads
388
Public
Repo size
1 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin501 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- roberta
- Access
- Public
- Created
- Jun 30, 2023
- Updated
- Jun 30, 2023
- SHA
- 9a57f833
Try a prompt
Model card
From the Hugging Face model README
- Baseline NLU model for the "AMUseBot" cooking taskbot prototype. Updated version with more robust req_ingredient intent recognition thanks to finetuning with extra synthetic data.
roberta-basemodel finetuned with default hyperparameters for 10 epochs on intents from the CookDial (https://github.com/YiweiJiang2015/CookDial) dataset with an extra choose_recipe intent added. Thesimpletransformerslibrary was used for fine-tuning.- Intent mapping: {"0": "affirm", "1": "choose_recipe", "2": "confirm", "3": "goodbye", "4": "greeting", "5": "negate", "6": "other", "7": "req_amount", "8": "req_duration", "9": "req_ingredient", "10": "req_ingredient_list", "11": "req_ingredient_list_ends", "12": "req_ingredient_list_length", "13": "req_instruction", "14": "req_is_recipe_finished", "15": "req_is_recipe_ongoing", "16": "req_parallel_action", "17": "req_repeat", "18": "req_start", "19": "req_substitute", "20": "req_temperature", "21": "req_title", "22": "req_tool", "23": "req_use_all", "24": "thank"}.
- Task
- Text Classification
- Library
- transformers
- Type
- roberta
- Languages
- en
- Created
- Jun 30, 2023
- Updated
- Jun 30, 2023