finiteautomata's picture
Update README.md
a2cc0f6 verified
metadata
language:
  - es
library_name: pysentimiento
pipeline_tag: text-classification
tags:
  - twitter
  - sentiment-analysis

Sentiment Analysis in Spanish

robertuito-sentiment-analysis

Repository: https://github.com/pysentimiento/pysentimiento/

Model trained with TASS 2020 corpus (around ~5k tweets) of several dialects of Spanish. Base model is RoBERTuito, a RoBERTa model trained in Spanish tweets.

Uses POS, NEG, NEU labels.

Usage

Use it directly with pysentimiento

from pysentimiento import create_analyzer
analyzer = create_analyzer(task="sentiment", lang="es")

analyzer.predict("Qué gran jugador es Messi")
# returns AnalyzerOutput(output=POS, probas={POS: 0.998, NEG: 0.002, NEU: 0.000})

Results

Results for the four tasks evaluated in pysentimiento. Results are expressed as Macro F1 scores

model emotion hate_speech irony sentiment
robertuito 0.560 ± 0.010 0.759 ± 0.007 0.739 ± 0.005 0.705 ± 0.003
roberta 0.527 ± 0.015 0.741 ± 0.012 0.721 ± 0.008 0.670 ± 0.006
bertin 0.524 ± 0.007 0.738 ± 0.007 0.713 ± 0.012 0.666 ± 0.005
beto_uncased 0.532 ± 0.012 0.727 ± 0.016 0.701 ± 0.007 0.651 ± 0.006
beto_cased 0.516 ± 0.012 0.724 ± 0.012 0.705 ± 0.009 0.662 ± 0.005
mbert_uncased 0.493 ± 0.010 0.718 ± 0.011 0.681 ± 0.010 0.617 ± 0.003
biGRU 0.264 ± 0.007 0.592 ± 0.018 0.631 ± 0.011 0.585 ± 0.011

Note that for Hate Speech, these are the results for Semeval 2019, Task 5 Subtask B

Citation

If you use this model in your research, please cite pysentimiento, RoBERTuito and TASS papers:


@article{perez2021pysentimiento,
  title={pysentimiento: a python toolkit for opinion mining and social NLP tasks},
  author={P{\'e}rez, Juan Manuel and Rajngewerc, Mariela and Giudici, Juan Carlos and Furman, Dami{\'a}n A and Luque, Franco and Alemany, Laura Alonso and Mart{\'\i}nez, Mar{\'\i}a Vanina},
  journal={arXiv preprint arXiv:2106.09462},
  year={2021}
}

@inproceedings{perez-etal-2022-robertuito,
    title = "{R}o{BERT}uito: a pre-trained language model for social media text in {S}panish",
    author = "P{\'e}rez, Juan Manuel  and
      Furman, Dami{\'a}n Ariel  and
      Alonso Alemany, Laura  and
      Luque, Franco M.",
    booktitle = "Proceedings of the Thirteenth Language Resources and Evaluation Conference",
    month = jun,
    year = "2022",
    address = "Marseille, France",
    publisher = "European Language Resources Association",
    url = "https://aclanthology.org/2022.lrec-1.785",
    pages = "7235--7243",
    abstract = "Since BERT appeared, Transformer language models and transfer learning have become state-of-the-art for natural language processing tasks. Recently, some works geared towards pre-training specially-crafted models for particular domains, such as scientific papers, medical documents, user-generated texts, among others. These domain-specific models have been shown to improve performance significantly in most tasks; however, for languages other than English, such models are not widely available. In this work, we present RoBERTuito, a pre-trained language model for user-generated text in Spanish, trained on over 500 million tweets. Experiments on a benchmark of tasks involving user-generated text showed that RoBERTuito outperformed other pre-trained language models in Spanish. In addition to this, our model has some cross-lingual abilities, achieving top results for English-Spanish tasks of the Linguistic Code-Switching Evaluation benchmark (LinCE) and also competitive performance against monolingual models in English Twitter tasks. To facilitate further research, we make RoBERTuito publicly available at the HuggingFace model hub together with the dataset used to pre-train it.",
}

@inproceedings{garcia2020overview,
  title={Overview of TASS 2020: Introducing emotion detection},
  author={Garc{\'\i}a-Vega, Manuel and D{\'\i}az-Galiano, MC and Garc{\'\i}a-Cumbreras, MA and Del Arco, FMP and Montejo-R{\'a}ez, A and Jim{\'e}nez-Zafra, SM and Mart{\'\i}nez C{\'a}mara, E and Aguilar, CA and Cabezudo, MAS and Chiruzzo, L and others},
  booktitle={Proceedings of the Iberian Languages Evaluation Forum (IberLEF 2020) Co-Located with 36th Conference of the Spanish Society for Natural Language Processing (SEPLN 2020), M{\'a}laga, Spain},
  pages={163--170},
  year={2020}
}