模型:

cardiffnlp/twitter-roberta-base-dec2021-tweet-topic-single-all

英文

cardiffnlp/twitter-roberta-base-dec2021-tweet-topic-single-all

这个模型是在 tweet_topic_single 上进行微调的版本 cardiffnlp/twitter-roberta-base-dec2021 。该模型使用train_all数据进行微调,并在test_2021数据上进行验证,用于tweet_topic任务。微调脚本可以在 here 中找到。在test_2021数据集上,它取得了以下结果:

  • F1(微平均):0.8948611931482575
  • F1(宏平均):0.800952410284692
  • 准确率:0.8948611931482575

用法

from transformers import pipeline

pipe = pipeline("text-classification", "cardiffnlp/twitter-roberta-base-dec2021-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"
}