cardiffnlp/tweet_topic_single
Viewer • Updated • 28.5k • 508 • 6
How to use cardiffnlp/roberta-base-tweet-topic-single-all with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="cardiffnlp/roberta-base-tweet-topic-single-all") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("cardiffnlp/roberta-base-tweet-topic-single-all")
model = AutoModelForSequenceClassification.from_pretrained("cardiffnlp/roberta-base-tweet-topic-single-all")This model is a fine-tuned version of roberta-base on the tweet_topic_single. This model is fine-tuned on train_all split and validated on test_2021 split of tweet_topic.
Fine-tuning script can be found here. It achieves the following results on the test_2021 set:
from transformers import pipeline
pipe = pipeline("text-classification", "cardiffnlp/roberta-base-tweet-topic-single-all")
topic = pipe("Love to take night time bike rides at the jersey shore. Seaside Heights boardwalk. Beautiful weather. Wishing everyone a safe Labor Day weekend in the US.")
print(topic)
@inproceedings{dimosthenis-etal-2022-twitter,
title = "{T}witter {T}opic {C}lassification",
author = "Antypas, Dimosthenis and
Ushio, Asahi and
Camacho-Collados, Jose and
Neves, Leonardo and
Silva, Vitor and
Barbieri, Francesco",
booktitle = "Proceedings of the 29th International Conference on Computational Linguistics",
month = oct,
year = "2022",
address = "Gyeongju, Republic of Korea",
publisher = "International Committee on Computational Linguistics"
}