TWEET-FID: An Annotated Dataset for Multiple Foodborne Illness Detection Tasks
Paper • 2205.10726 • Published • 1
How to use auro736/xlm-roberta-large-tweet-fid-EMD with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="auro736/xlm-roberta-large-tweet-fid-EMD") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, ModelForTokenClassificationWithCRF
tokenizer = AutoTokenizer.from_pretrained("auro736/xlm-roberta-large-tweet-fid-EMD")
model = ModelForTokenClassificationWithCRF.from_pretrained("auro736/xlm-roberta-large-tweet-fid-EMD", device_map="auto")This is a XLM-RoBERTa-large model trained on the Tweet-FID dataset ("TWEET-FID: An Annotated Dataset for Multiple Foodborne Illness Detection Tasks", Ruofan Hu et al, 2022 ) which is a collection of Twitter to detect incidents of foodborne illnesses.
The model is enriched with a multi class classification head to perform the custom task called Entity Mention Detection (EMD). The objective is to determine predefined entities (food, location, symptom, other) in a given text related to a food risk.