Hugging Face's logo

kedudzic
/
roberta-base-cookdial

Text Classification
Transformers
PyTorch
English
roberta
Inference Endpoints
Model card Files Community
Edit model card
  • Baseline NLU model for the "AMUseBot" cooking taskbot prototype.
  • roberta-base model 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. The simpletransformers library 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"}.
Downloads last month
7
Inference Examples
Text Classification
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.