cardiffnlp/roberta-base-emotion

该模型是通过对 tweet_eval (emotion) 进行微调得到的版本，使用了 tweetnlp 进行训练。训练集采用的是train，并且参数在验证集validation上进行了调整。

在测试集test（ link ）上达到了以下指标。

• F1（micro）：0.8198451794510907
• F1（macro）：0.7858041485997908
• 准确率：0.8198451794510908

用法

通过pip安装tweetnlp。

pip install tweetnlp

在Python中加载模型。

import tweetnlp
model = tweetnlp.Classifier("cardiffnlp/roberta-base-emotion", max_length=128)
model.predict('Get the all-analog Classic Vinyl Edition of "Takin Off" Album from {@herbiehancock@} via {@bluenoterecords@} link below {{URL}}')

参考资料

@inproceedings{camacho-collados-etal-2022-tweetnlp,
    title={{T}weet{NLP}: {C}utting-{E}dge {N}atural {L}anguage {P}rocessing for {S}ocial {M}edia},
    author={Camacho-Collados, Jose and Rezaee, Kiamehr and Riahi, Talayeh and Ushio, Asahi and Loureiro, Daniel and Antypas, Dimosthenis and Boisson, Joanne and Espinosa-Anke, Luis and Liu, Fangyu and Mart{'\i}nez-C{'a}mara, Eugenio and others},
    author = "Ushio, Asahi  and
      Camacho-Collados, Jose",
    booktitle = "Proceedings of the 2022 Conference on Empirical Methods in Natural Language Processing: System Demonstrations",
    month = nov,
    year = "2022",
    address = "Abu Dhabi, U.A.E.",
    publisher = "Association for Computational Linguistics",
}