CoolFace
Modelpublic

LEIA/LEIA-base

sourceHugging Facemitupdated 2y agoView on Hugging Face
1likes8downloads
README.md29 linesDownload Raw Back to root
1---2license: mit3language:4- en5library_name: transformers6pipeline_tag: text-classification7widget:8- text: "You wont believe what happened to me today"9- text: "You wont believe what happened to me today!"10- text: "You wont believe what happened to me today..."11- text: "You wont believe what happened to me today <3"12- text: "You wont believe what happened to me today :)"13- text: "You wont believe what happened to me today :("14---15This is an emotion classification model based on further pre-training of BERTweet-base with preferential masking of emotion words and fine-tuning on a subset of a self-labeled emotion dataset (Lykousas et al., 2019) that corresponds to Anger, Fear, Sadness, Joy, and Affection. The paper, [LEIA: Linguistic Embeddings for the Identification of Affect](https://doi.org/10.1140/epjds/s13688-023-00427-0) provides further details on the model and its evauation. 16 17See [LEIA-large](https://huggingface.co/LEIA/LEIA-large) for a similar model based on BERTweet-large.18## Citation19Please cite the following paper if you find the model useful for your work:20```bibtex21@article{aroyehun2023leia,22  title={LEIA: Linguistic Embeddings for the Identification of Affect},23  author={Aroyehun, Segun Taofeek and Malik, Lukas and Metzler, Hannah and Haimerl, Nikolas and Di Natale, Anna and Garcia, David},24  journal={EPJ Data Science},25  volume={12},26  year={2023},27  publisher={Springer}28}29```